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