PatchTST_ad.yaml 514 B

1234567891011121314151617181920212223242526272829303132
  1. batch_size: 128
  2. seq_len: 100
  3. epoch: 5
  4. training: True
  5. do_eval: True
  6. task: anomaly
  7. to_static_train: False
  8. use_amp: False
  9. amp_level: O2
  10. dataset:
  11. name: TSADDataset
  12. dataset_root: ./data/
  13. train_path: ./data/train.csv
  14. val_path: ./data/val.csv
  15. scale: True
  16. time_feat: False
  17. info_params:
  18. freq: 1
  19. label_col: "label"
  20. time_col: "timestamp"
  21. feature_cols: "feature_0,feature_1"
  22. model:
  23. name: PatchTST_AD
  24. model_cfg:
  25. optimizer_params:
  26. learning_rate: 0.001
  27. gamma: 0.5
  28. output: 'output/'