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

update hyperparameters

parent f19e2e0a
No related merge requests found
......@@ -77,7 +77,6 @@ class Class1NeuralNetwork(object):
fit_hyperparameter_defaults = HyperparameterDefaults(
max_epochs=500,
take_best_epoch=False, # currently unused
validation_split=0.1,
early_stopping=True,
minibatch_size=128,
......@@ -92,11 +91,7 @@ class Class1NeuralNetwork(object):
"""
early_stopping_hyperparameter_defaults = HyperparameterDefaults(
patience=10,
monitor='val_loss', # currently unused
min_delta=0, # currently unused
verbose=1, # currently unused
mode='auto' # currently unused
patience=20,
)
"""
Hyperparameters for early stopping.
......@@ -129,6 +124,11 @@ class Class1NeuralNetwork(object):
hyperparameter_renames = {
"use_embedding": None,
"pseudosequence_use_embedding": None,
"monitor": None,
"min_delta": None,
"verbose": None,
"mode": None,
"take_best_epoch": None,
}
@classmethod
......
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