TimesNet.yaml 692 B

123456789101112131415161718192021222324252627282930313233
  1. Global:
  2. model: TimesNet
  3. mode: check_dataset # check_dataset/train/evaluate/predict
  4. dataset_dir: "/paddle/dataset/paddlex/ts_fc/ts_dataset_examples"
  5. device: gpu:0
  6. output: "output"
  7. CheckDataset:
  8. convert:
  9. enable: False
  10. src_dataset_type: nul
  11. split:
  12. enable: False
  13. train_percent: null
  14. val_percent: null
  15. Train:
  16. time_col: date
  17. target_cols: OT
  18. freq: 1h
  19. input_len: 96
  20. predict_len: 96
  21. epochs_iters: 40
  22. batch_size: 16
  23. learning_rate: 0.0001
  24. patience: 10
  25. Evaluate:
  26. weight_path: "output/best_accuracy.pdparams.tar"
  27. Predict:
  28. model_dir: "output/best_model/model.pdparams"
  29. input_path: "/paddle/dataset/paddlex/ts_fc/ts_dataset_examples/test.csv"