TiDE.yaml 625 B

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. batch_size: 32
  2. seq_len: 720
  3. predict_len: 96
  4. do_eval: True
  5. epoch: 2
  6. to_static_train: False
  7. use_amp: False
  8. amp_level: O2
  9. dataset:
  10. name: TSDataset
  11. dataset_root: /data/
  12. train_path: /data/train.csv
  13. val_path: /data/val.csv
  14. time_feat: True
  15. use_holiday: True
  16. scale: True
  17. info_params:
  18. freq: 1h
  19. target_cols: "OT"
  20. time_col: date
  21. model:
  22. name: TiDE
  23. model_cfg:
  24. c_in: 1
  25. num_encoder_layers: 2
  26. use_revin: False
  27. drop_prob: 0.5
  28. hidden_size: 1024
  29. decoder_output_dim: 8
  30. temporal_decoder_hidden: 64
  31. optimizer_params:
  32. learning_rate: 0.00099
  33. gamma: 0.5
  34. patience: 10