Skip to content
Snippets Groups Projects
Commit 20d68c2b authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

add learning rate hyperparameter

parent a9cc8176
No related merge requests found
......@@ -61,6 +61,7 @@ class Class1NeuralNetwork(object):
compile_hyperparameter_defaults = HyperparameterDefaults(
loss="custom:mse_with_inequalities",
optimizer="rmsprop",
learning_rate=None,
)
"""
Loss and optimizer hyperparameters. Any values supported by keras may be
......@@ -575,6 +576,12 @@ class Class1NeuralNetwork(object):
loss=loss_name_or_function,
optimizer=self.hyperparameters['optimizer'])
if self.hyperparameters['learning_rate'] is not None:
from keras import backend as K
K.set_value(
self.network().optimizer.lr,
self.hyperparameters['learning_rate'])
if loss_supports_inequalities:
# Do not sample negative affinities: just use an inequality.
random_negative_ic50 = self.hyperparameters['random_negative_affinity_min']
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment