TrafficWheel/config/ARIMA/PEMSD7.yaml

47 lines
693 B
YAML

data:
add_day_in_week: true
add_time_in_day: true
column_wise: false
days_per_week: 7
default_graph: true
horizon: 12
lag: 12
normalizer: std
num_nodes: 883
steps_per_day: 288
test_ratio: 0.2
tod: false
val_ratio: 0.2
log:
log_step: 3000
plot: false
model:
input_dim: 1
output_dim: 1
p: 2
d: 1
q: 1
drift: True
test:
mae_thresh: None
mape_thresh: 0.0
train:
batch_size: 16
early_stop: true
early_stop_patience: 10
epochs: 200
grad_norm: false
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- '5'
- '20'
- '40'
- '70'
lr_init: 0.00075
max_grad_norm: 5
real_value: true
seed: 10
weight_decay: 0