PatchTST.yaml 598 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. batch_size: 32
  2. seq_len: 96
  3. predict_len: 96
  4. do_eval: True
  5. epoch: 5
  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: False
  15. scale: True
  16. info_params:
  17. freq: 1h
  18. target_cols: "OT"
  19. time_col: date
  20. model:
  21. name: PatchTSTModel
  22. model_cfg:
  23. c_in: 7
  24. n_layers: 3
  25. n_heads: 4
  26. d_model: 16
  27. d_ff: 128
  28. dropout: 0.3
  29. fc_dropout: 0.3
  30. head_dropout: 0.0
  31. patch_len: 16
  32. stride: 8
  33. patience: 20
  34. optimizer_params:
  35. learning_rate: 0.0001