jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-02 23:41:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

add early stopping in config

Browse Source
This commit is contained in:
Jamie
2021-05-10 21:42:39 -04:00
parent ab2841bd01
commit aa1b9032fb
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gamestonk_terminal/config_neural_network_models.py
View File

@@ -48,3 +48,6 @@ Convolutional = [
{"Flatten": {}},
{"Dense": {"activation": "relu"}},
]
# Add early stopping to save time training when validation is plateaued
Early_Stop_Patience = 10

7
gamestonk_terminal/prediction_techniques/neural_networks_view.py
View File

@@ -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