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

Saving samples plot as png at end of training

Browse Source
This commit is contained in:
Victor Mylle
2024-04-19 14:05:20 +02:00
parent 4e713ef564
commit 46c7c6f7e5
4 changed files with 60 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/training_scripts/autoregressive_quantiles.py
View File

@@ -2,7 +2,9 @@ from src.utils.clearml import ClearMLHelper
#### ClearML ####
clearml_helper = ClearMLHelper(project_name="Thesis/NrvForecast")
task = clearml_helper.get_task(task_name="AQR: Linear Baseline + Quarter Trigonometric")
task = clearml_helper.get_task(
task_name="AQR: Linear Baseline + Load + PV + Wind + Net Position + Quarter"
)
task.execute_remotely(queue_name="default", exit_process=True)
from src.policies.PolicyEvaluator import PolicyEvaluator
@@ -70,16 +72,16 @@ model_parameters = {
"hidden_size": 256,
"num_layers": 2,
"dropout": 0.2,
"time_feature_embedding": 2,
"time_feature_embedding": 5,
}
model_parameters = task.connect(model_parameters, name="model_parameters")
# time_embedding = TimeEmbedding(
# data_processor.get_time_feature_size(), model_parameters["time_feature_embedding"]
# )
time_embedding = TimeEmbedding(
data_processor.get_time_feature_size(), model_parameters["time_feature_embedding"]
)
time_embedding = TrigonometricTimeEmbedding(data_processor.get_time_feature_size())
# time_embedding = TrigonometricTimeEmbedding(data_processor.get_time_feature_size())
# lstm_model = GRUModel(
# time_embedding.output_dim(inputDim),