basic:
  dataset: "PEMSD4"
  mode: "train"
  device: "cuda:0"
  model: "STID"

data:
  num_nodes: 307
  lag: 12
  horizon: 12
  val_ratio: 0.2
  test_ratio: 0.2
  tod: False
  normalizer: std
  column_wise: False
  default_graph: True
  add_time_in_day: True
  add_day_in_week: True
  steps_per_day: 288
  days_per_week: 7

model:
  input_dim: 3
  output_dim: 1
  history: 12
  horizon: 12
  num_nodes: 307
  input_len: 12
  embed_dim": 32
  output_len: 12
  num_layer: 3
  if_node: True
  node_dim: 32
  if_T_i_D: True
  if_D_i_W: True
  temp_dim_tid: 32
  temp_dim_diw: 32
  time_of_day_size: 288
  day_of_week_size: 7


train:
  loss_func: mae
  seed: 1
  batch_size: 64
  epochs: 300
  lr_init: 0.002
  weight_decay: 0.0001
  lr_decay: False
  lr_decay_rate: 0.3
  lr_decay_step: "1,50,80"
  early_stop: True
  early_stop_patience: 15
  grad_norm: False
  max_grad_norm: 5
  real_value: True

test:
  mae_thresh: null
  mape_thresh: 0.0

log:
  log_step: 200
  plot: False