TimesNet_ad.yaml 557 B

12345678910111213141516171819202122232425262728293031323334
  1. batch_size: 128
  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: TimesNet_AD
  24. model_cfg:
  25. c_in: 2
  26. e_layers: 1
  27. num_kernels: 6
  28. d_model: 8
  29. optimizer_params:
  30. learning_rate: 0.001
  31. gamma: 0.5