diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py index 6cf42fcc930b2a2206cf655ceb3c3c491d710006..5a1f6d563dd2dd37b2da1258bcf73ffe9e038195 100644 --- a/mhcflurry/class1_neural_network.py +++ b/mhcflurry/class1_neural_network.py @@ -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