Skip to content

Commit

Permalink
Change load module & added new folder
Browse files Browse the repository at this point in the history
  • Loading branch information
fornasari12 committed Dec 2, 2021
1 parent 0c9e474 commit 872aeba
Show file tree
Hide file tree
Showing 832 changed files with 373,954 additions and 5 deletions.
9 changes: 5 additions & 4 deletions config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,8 @@ general:
'realTraffic',
'realKnownCause',
'realAdExchange',
'realAWSCloudWatch',
'realAWSCloudwatch',
# 'M4Hourly'
]
scaler_path_nbeats: "/Volumes/GoogleDrive/My Drive/Colab_Notebooks/model/n_beats/scaler.pickle"
model_path_nbeats: "/Volumes/GoogleDrive/My Drive/Colab_Notebooks/model/n_beats/n_beats.pickle"
Expand All @@ -21,10 +22,10 @@ model:
"weekofyear"
]
time_varying_known_reals: [ "time_idx" ]
batch_size: 1024
max_epochs: 3000
batch_size: 512
max_epochs: 600
gpus: 1
learning_rate: 0.001
learning_rate: 0.005
hidden_size: 100
dropout: 0.21387145479502692
hidden_continuous_size: 50
Expand Down
Loading

0 comments on commit 872aeba

Please sign in to comment.