From a9cc81763803e52310eca7d62504ab12f18ecf80 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Wed, 14 Feb 2018 11:24:48 -0500
Subject: [PATCH] update hyperparameters
---
mhcflurry/class1_neural_network.py | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 6cf42fcc..5a1f6d56 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
--
GitLab