Updated thesis
This commit is contained in:
@@ -5,7 +5,7 @@ clearml_helper = ClearMLHelper(
|
||||
project_name="Thesis/NrvForecast"
|
||||
)
|
||||
task = clearml_helper.get_task(
|
||||
task_name="NAQR: Non Linear (2 - 512)"
|
||||
task_name="NAQR: Non Linear (2 - 256)"
|
||||
)
|
||||
task.execute_remotely(queue_name="default", exit_process=True)
|
||||
|
||||
@@ -30,7 +30,7 @@ from src.models.time_embedding_layer import TimeEmbedding
|
||||
#### Data Processor ####
|
||||
data_config = DataConfig()
|
||||
|
||||
data_config.NRV_HISTORY = False
|
||||
data_config.NRV_HISTORY = True
|
||||
data_config.LOAD_HISTORY = False
|
||||
data_config.LOAD_FORECAST = False
|
||||
|
||||
@@ -53,7 +53,7 @@ data_processor.set_full_day_skip(True)
|
||||
#### Hyperparameters ####
|
||||
data_processor.set_output_size(96)
|
||||
inputDim = data_processor.get_input_size()
|
||||
epochs = 5
|
||||
epochs = 300
|
||||
|
||||
# add parameters to clearml
|
||||
quantiles = task.get_parameter("general/quantiles", cast=True)
|
||||
@@ -68,7 +68,7 @@ else:
|
||||
|
||||
model_parameters = {
|
||||
"learning_rate": 0.0001,
|
||||
"hidden_size": 512,
|
||||
"hidden_size": 256,
|
||||
"num_layers": 2,
|
||||
"dropout": 0.2,
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user