DLinear.yaml 489 B

123456789101112131415161718192021222324252627282930
  1. batch_size: 16
  2. seq_len: 96
  3. predict_len: 336
  4. sampling_stride: 1
  5. do_eval: True
  6. epoch: 10
  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: DLinearModel
  21. model_cfg:
  22. c_in: 321
  23. individual: False
  24. optimizer_params:
  25. learning_rate: 0.001
  26. gamma: 0.5
  27. patience: 3
  28. test:
  29. stride: 1