config.yaml 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. # jianding/config.yaml
  2. project:
  3. plant_name: "jianding"
  4. files:
  5. dataset_dir: "datasets"
  6. file_pattern: "data_process_{}.csv"
  7. model_filename: "model.pth"
  8. scaler_filename: "scaler.pkl"
  9. output_csv_filename: "predictions.csv"
  10. edge_index_filename: "edge_index.pt"
  11. data_split:
  12. start_files: 1
  13. end_files: 24
  14. train_start_date: "2024-10-08"
  15. train_end_date: "2026-02-13"
  16. val_start_date: "2024-10-08"
  17. val_end_date: "2026-02-13"
  18. test_start_date: "2024-10-08"
  19. test_end_date: "2026-02-13"
  20. model_params:
  21. seq_len: 10
  22. output_size: 5
  23. step_size: 5
  24. resolution: 60
  25. hidden_size: 64
  26. num_layers: 1
  27. dropout: 0.0
  28. training_params:
  29. epochs: 200
  30. lr: 0.01
  31. batch_size: 512
  32. scheduler_step_size: 100
  33. scheduler_gamma: 0.9
  34. patience: 200
  35. min_delta: 1.0e-10
  36. device: 1 # 对应 args.device
  37. random_seed: 1314
  38. sensors:
  39. # 输入传感器列
  40. required_columns:
  41. - "index"
  42. - "water_out"
  43. - "ns=3;s=AI_ROJSLL_OUT"
  44. - "ns=3;s=AI_UFCSLL_OUT"
  45. - "ns=3;s=RO_1DJSLL_SSD"
  46. - "ns=3;s=RO_2DJSLL_SSD"
  47. - "ns=3;s=RO_NS_SSD"
  48. - "ns=3;s=AI_JYCSLL1_OUT"
  49. - "ns=3;s=AI_RODJYL_OUT"
  50. - "ns=3;s=AI_ROJSYL_OUT"
  51. - "ns=3;s=AI_UFCSYL_OUT"
  52. - "ns=3;s=AI_JYCIPPH_OUT"
  53. - "ns=3;s=AI_JYCSDD_OUT"
  54. - "ns=3;s=AI_UFCSZD_OUT"
  55. - "ns=3;s=AI_ROCSDD_OUT"
  56. - "ns=3;s=AI_UFJSORP_OUT"
  57. - "ns=3;s=AI_UFJSPH_OUT"
  58. - "ns=3;s=AI_UFJSYW_OUT"
  59. - "ns=3;s=AI_JYROCSYW_OUT"
  60. - "ns=3;s=AI_JYSYW_OUT"
  61. - "ns=3;s=AI_RODJB_FR_OUT"
  62. - "ns=3;s=AI_ROGSB_FR_OUT"
  63. - "ns=3;s=AI_ROGYB_FR_OUT"
  64. - "ns=3;s=AI_UFFXB_FR_OUT"
  65. - "ns=3;s=AI_UFCSB_FR_OUT"
  66. - "ns=3;s=UF_TMP"
  67. - "ns=3;s=RO_CHA1YL_SSD"
  68. - "ns=3;s=RO_CHA2YL_SSD"
  69. - "ns=3;s=RO_ZCS_SSD"
  70. # 最终预测目标列
  71. target_columns:
  72. - "ns=3;s=UF_TMP"
  73. - "ns=3;s=RO_CHA1YL_SSD"
  74. - "ns=3;s=RO_CHA2YL_SSD"
  75. - "ns=3;s=RO_ZCS_SSD"