diff --git a/.gitattributes b/.gitattributes deleted file mode 100644 index 87e654b..0000000 --- a/.gitattributes +++ /dev/null @@ -1 +0,0 @@ -*.csv filter=lfs diff=lfs merge=lfs -text diff --git a/src/training_scripts/autoregressive_quantiles.py b/src/training_scripts/autoregressive_quantiles.py index 1fac750..a79b85d 100644 --- a/src/training_scripts/autoregressive_quantiles.py +++ b/src/training_scripts/autoregressive_quantiles.py @@ -59,11 +59,11 @@ else: quantiles = eval(quantiles) model_parameters = { - "learning_rate": 0.001, - "hidden_size": 512, + "learning_rate": 0.0001, + "hidden_size": 256, "num_layers": 4, "dropout": 0.2, - "time_feature_embedding": 8, + "time_feature_embedding": 16, } model_parameters = task.connect(model_parameters, name="model_parameters") @@ -96,7 +96,7 @@ trainer = AutoRegressiveQuantileTrainer( trainer.add_metrics_to_track( [PinballLoss(quantiles), MSELoss(), L1Loss(), CRPSLoss(quantiles)] ) -trainer.early_stopping(patience=15) +trainer.early_stopping(patience=30) trainer.plot_every(5) trainer.train(task=task, epochs=epochs, remotely=True)