NLinear.yaml 840 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. Global:
  2. model: NLinear
  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: null
  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. log_interval: 10
  26. Evaluate:
  27. weight_path: "output/best_model/model.pdparams"
  28. Export:
  29. weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/NLinear.pdparams.tar
  30. Predict:
  31. batch_size: 1
  32. model_dir: "output/inference"
  33. input: "https://paddle-model-ecology.bj.bcebos.com/paddlex/ts/demo_ts/ts_fc.csv"