Nonstationary.yaml 516 B

123456789101112131415161718192021222324252627282930313233
  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: True
  15. scale: True
  16. info_params:
  17. freq: 1h
  18. target_cols: "OT"
  19. time_col: date
  20. model:
  21. name: Nonstationary_Transformer
  22. model_cfg:
  23. c_in: 7
  24. factor: 1
  25. p_hidden_dims: [256, 256]
  26. optimizer_params:
  27. learning_rate: 0.0001
  28. gamma: 0.5
  29. patience: 3