VictorMylle/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rewrote the NRVDataset to be cleaner

Browse Source
This commit is contained in:
Victor Mylle
2023-11-28 15:35:35 +00:00
parent f9e8f9e69f
commit ffa19592f9
3 changed files with 83 additions and 181 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/training_scripts/autoregressive_quantiles.py
View File

@@ -16,7 +16,7 @@ from src.models.time_embedding_layer import TimeEmbedding
#### ClearML ####
clearml_helper = ClearMLHelper(project_name="Thesis/NrvForecast")
task = clearml_helper.get_task(task_name="None")
task = clearml_helper.get_task(task_name="Autoregressive Quantile Regression")
#### Data Processor ####
@@ -63,8 +63,8 @@ trainer = AutoRegressiveQuantileTrainer(
debug=True,
)
trainer.add_metrics_to_track(
[PinballLoss(quantiles), MSELoss(), L1Loss(), CRPSLoss(quantiles)]
[PinballLoss(quantiles), MSELoss(), L1Loss(), CRPSLoss()]
)
trainer.early_stopping(patience=10)
trainer.plot_every(5)
trainer.train(task=task, epochs=epochs, remotely=True)
trainer.train(task=task, epochs=epochs, remotely=False)