NLinear.yaml 531 B

12345678910111213141516171819202122232425262728293031323334353637
  1. batch_size: 16
  2. seq_len: 96
  3. predict_len: 336
  4. sampling_stride: 1
  5. do_eval: True
  6. epoch: 5
  7. to_static_train: False
  8. use_amp: False
  9. amp_level: O2
  10. dataset:
  11. name: TSDataset
  12. dataset_root: /data/
  13. train_path: /data/train.csv
  14. val_path: /data/val.csv
  15. time_feat: False
  16. scale: True
  17. info_params:
  18. time_col: date
  19. model:
  20. name: NLinearModel
  21. model_cfg:
  22. c_in: 321
  23. individual: True
  24. optimizer_params:
  25. learning_rate: 0.005
  26. gamma: 0.5
  27. patience: 3
  28. #pretrain: NLinear_init.pdparams
  29. test:
  30. stride: 1