TimesNet_cls.yaml 704 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. batch_size: 16 #
  2. seq_len: 96 #
  3. predict_len: 0 #
  4. label_len: 0 #
  5. do_eval: True #
  6. epoch: 30 # max_epochs
  7. training: True #
  8. eval_metrics: ['acc', ]
  9. task: classification
  10. dataset:
  11. name: TSCLSDataset
  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. time_col: time
  19. group_id: group_id
  20. target_cols: ['dim_0', 'dim_1', 'dim_2']
  21. static_cov_cols: label
  22. model:
  23. name: TimesNet_CLS
  24. model_cfg:
  25. e_layers: 3 #
  26. num_kernels: 6 #
  27. d_model: 32 #
  28. d_ff: 32 #
  29. top_k: 3 #
  30. window_sampling_limit: Null #
  31. patience: 10 #
  32. optimizer_params:
  33. learning_rate: 0.001
  34. gamma: 0.9
  35. output: 'output/'