-
Notifications
You must be signed in to change notification settings - Fork 3
/
conf.etdataset.yml
52 lines (36 loc) · 1.11 KB
/
conf.etdataset.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
---
# REQUIRED: Output directory
output_dir: output.etdataset
# REQUIRED: Input length
input_length: 512
# REQUIRED: Prediction length
prediction_length: 96
# REQUIRED: Number of features
no_features: 7
# REQUIRED: Number of mixer layers
no_mixer_layers: 2
# REQUIRED: Data source
data_src: csv-file
# REQUIRED: Path to the data source
data_src_csv: data/ETTh1.csv
# Optional: How to initialize the model (restarts training from scratch if set to from-scratch)
initialize: from-scratch # from-best-checkpoint
# Optional: Batch size
batch_size: 32
# Optional: Number of epochs
num_epochs: 100
# Optional: Learning rate
learning_rate: 0.00001
# Optional: Optimizer - Adam, SGD, RMSprop, etc
optimizer: Adam
# Optional: Random seed
random_seed: 42
# Optional: Validation split method
validation_split: temporal-holdout
# Optional: Validation split holdout - fraction of the data to be used for validation
# (only used if validation_split is set to temporal-holdout)
validation_split_holdout: 0.2
# Optional: dropout
dropout: 0.3
# Optional: Number of hidden channels in the feature mixing layer
feat_mixing_hidden_channels: null