From aa1b9032fb5950ef06171e90f6b53b1ab49fccb5 Mon Sep 17 00:00:00 2001 From: Jamie Date: Mon, 10 May 2021 21:42:39 -0400 Subject: [PATCH] add early stopping in config --- gamestonk_terminal/config_neural_network_models.py | 3 +++ .../prediction_techniques/neural_networks_view.py | 7 +++++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/gamestonk_terminal/config_neural_network_models.py b/gamestonk_terminal/config_neural_network_models.py index 28c7181e870..30b8af07831 100644 --- a/gamestonk_terminal/config_neural_network_models.py +++ b/gamestonk_terminal/config_neural_network_models.py @@ -48,3 +48,6 @@ Convolutional = [ {"Flatten": {}}, {"Dense": {"activation": "relu"}}, ] + +# Add early stopping to save time training when validation is plateaued +Early_Stop_Patience = 10 \ No newline at end of file diff --git a/gamestonk_terminal/prediction_techniques/neural_networks_view.py b/gamestonk_terminal/prediction_techniques/neural_networks_view.py index ba41c4988e8..f12d159d63c 100644 --- a/gamestonk_terminal/prediction_techniques/neural_networks_view.py +++ b/gamestonk_terminal/prediction_techniques/neural_networks_view.py @@ -28,8 +28,11 @@ from gamestonk_terminal.prediction_techniques.pred_helper import ( ) from gamestonk_terminal import config_neural_network_models as cfg_nn_models -es = EarlyStopping(monitor="val_loss", patience=10) - +if cfg_nn_models.Early_Stop_Patience: + es = EarlyStopping(monitor="val_loss", patience=cfg_nn_models.Early_Stop_Patience) +else: + #Set patience to very long value + es = EarlyStopping(monitor="val_loss", patience=1000) def build_neural_network_model( Recurrent_Neural_Network: List[Any], n_inputs: int, n_days: int