37 lines
763 B
JSON
37 lines
763 B
JSON
{
|
|
"input_size": 7,
|
|
"output_size": 1,
|
|
"hidden_size": 512,
|
|
"n_stacks_trend": 3,
|
|
"n_stacks_seasonal": 4,
|
|
"blocks_per_stack": 3,
|
|
"theta_size_trend": 2,
|
|
"theta_size_season": 4,
|
|
"epochs": 300,
|
|
"batch_size": 32,
|
|
"learning_rate": 0.0001,
|
|
"loss_type": "mse",
|
|
"alpha": 0.1,
|
|
"neuron_function": "GELU(approximate='none')",
|
|
"device": "cuda",
|
|
"train_samples": 1971,
|
|
"validation_samples": 494,
|
|
"train_input_shape": [
|
|
1971,
|
|
7
|
|
],
|
|
"train_target_shape": [
|
|
1971,
|
|
1
|
|
],
|
|
"validation_input_shape": [
|
|
494,
|
|
7
|
|
],
|
|
"validation_target_shape": [
|
|
494,
|
|
1
|
|
],
|
|
"best_epoch": 300,
|
|
"best_validation_loss": 4.962215825798921e-05
|
|
} |