Nonstationary_ad.yaml 582 B

1234567891011121314151617181920212223242526272829303132333435
  1. batch_size: 32
  2. seq_len: 100
  3. do_eval: True
  4. epoch: 5
  5. training: 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: NonStationary_AD
  24. model_cfg:
  25. c_in: 5
  26. factor: 3
  27. p_hidden_dims: [32, 32]
  28. d_model: 64
  29. d_ff: 64
  30. optimizer_params:
  31. learning_rate: 0.001
  32. gamma: 0.5