config.yaml 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142
  1. # xishan_config.yaml
  2. project:
  3. plant_name: "xishan"
  4. files:
  5. dataset_dir: "datasets_xishan"
  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: 68
  14. train_start_date: "2024-02-23"
  15. train_end_date: "2026-05-14"
  16. val_start_date: "2024-02-23"
  17. val_end_date: "2026-05-14"
  18. test_start_date: "2024-02-23"
  19. test_end_date: "2026-05-14"
  20. model_params:
  21. seq_len: 10
  22. output_size: 5
  23. step_size: 5
  24. resolution: 60
  25. feature_num: 79
  26. labels_num: 16
  27. hidden_size: 64
  28. num_layers: 1
  29. dropout: 0.0
  30. training_params:
  31. epochs: 200
  32. lr: 0.01
  33. batch_size: 1024
  34. scheduler_step_size: 100
  35. scheduler_gamma: 0.9
  36. patience: 200
  37. min_delta: 1.0e-10
  38. device: 0
  39. random_seed: 1314
  40. sensors:
  41. # 输入传感器列
  42. required_columns:
  43. - "index"
  44. - "C.M.FT_ZGJJY1@out"
  45. - "C.M.RO1_FT_JS@out"
  46. - "C.M.RO2_FT_JS@out"
  47. - "C.M.RO3_FT_JS@out"
  48. - "C.M.RO4_FT_JS@out"
  49. - "C.M.UF1_FT_JS@out"
  50. - "C.M.UF2_FT_JS@out"
  51. - "C.M.UF3_FT_JS@out"
  52. - "C.M.UF4_FT_JS@out"
  53. - "C.M.UF_FT_ZCS@out"
  54. - "C.M.FT_ZGJJY2@out"
  55. - "C.M.FT_ZGJJY3@out"
  56. - "C.M.FT_ZGJJY4@out"
  57. - "C.M.RO1_PT_JS@out"
  58. - "C.M.RO2_PT_JS@out"
  59. - "C.M.RO3_PT_JS@out"
  60. - "C.M.UF1_PT_JS@out"
  61. - "C.M.UF2_PT_JS@out"
  62. - "C.M.UF3_PT_JS@out"
  63. - "C.M.UF4_PT_JS@out"
  64. - "C.M.LT_JSC@out"
  65. - "C.M.RO1_PT_CS@out"
  66. - "C.M.RO1_PT_DJ2@out"
  67. - "C.M.RO2_PT_CS@out"
  68. - "C.M.RO2_PT_DJ2@out"
  69. - "C.M.RO3_PT_CS@out"
  70. - "C.M.RO3_PT_DJ2@out"
  71. - "C.M.RO4_PT_CS@out"
  72. - "C.M.RO4_PT_DJ2@out"
  73. - "C.M.RO4_PT_JS@out"
  74. - "C.M.LT_HCl@out"
  75. - "C.M.LT_NaClO@out"
  76. - "C.M.LT_PAC@out"
  77. - "C.M.LT_QSC@out"
  78. - "C.M.RO_Cond_ZCS@out"
  79. - "C.M.RO_TT_ZJS@out"
  80. - "C.M.UF1_JSF_kd@out"
  81. - "C.M.UF2_JSF_kd@out"
  82. - "C.M.UF_GSB4_fre@out"
  83. - "C.M.UF_ORP_ZCS@out"
  84. - "C.M.JYB2_ZGJ1_fre@out"
  85. - "C.M.JYB2_ZGJ2_fre@out"
  86. - "C.M.JYB2_ZGJ3_fre@out"
  87. - "C.M.JYB2_ZGJ4_fre@out"
  88. - "C.M.RO1_GYB_fre@out"
  89. - "C.M.RO2_GYB_fre@out"
  90. - "C.M.RO3_GYB_fre@out"
  91. - "C.M.RO4_GYB_fre@out"
  92. - "C.M.UF3_JSF_kd@out"
  93. - "C.M.UF4_JSF_kd@out"
  94. - "C.M.UF_FXB2_fre@out"
  95. - "C.M.RO1_DJB_fre@out"
  96. - "C.M.RO1_GYBF_kd@out"
  97. - "C.M.RO2_DJB_fre@out"
  98. - "C.M.RO2_GYBF_kd@out"
  99. - "C.M.RO3_DJB_fre@out"
  100. - "C.M.RO3_GYBF_kd@out"
  101. - "C.M.RO4_DJB_fre@out"
  102. - "C.M.RO4_GYBF_kd@out"
  103. - "C.M.UF1_DB@press_PV"
  104. - "C.M.UF2_DB@press_PV"
  105. - "C.M.UF3_DB@press_PV"
  106. - "C.M.UF4_DB@press_PV"
  107. - "UF1Per"
  108. - "UF2Per"
  109. - "UF3Per"
  110. - "UF4Per"
  111. - "C.M.RO1_DB@DPT_1"
  112. - "C.M.RO2_DB@DPT_1"
  113. - "C.M.RO3_DB@DPT_1"
  114. - "C.M.RO4_DB@DPT_1"
  115. - "C.M.RO1_DB@DPT_2"
  116. - "C.M.RO2_DB@DPT_2"
  117. - "C.M.RO3_DB@DPT_2"
  118. - "C.M.RO4_DB@DPT_2"
  119. # 最终预测目标列
  120. target_columns:
  121. - "C.M.UF1_DB@press_PV"
  122. - "C.M.UF2_DB@press_PV"
  123. - "C.M.UF3_DB@press_PV"
  124. - "C.M.UF4_DB@press_PV"
  125. - "UF1Per"
  126. - "UF2Per"
  127. - "UF3Per"
  128. - "UF4Per"
  129. - "C.M.RO1_DB@DPT_1"
  130. - "C.M.RO2_DB@DPT_1"
  131. - "C.M.RO3_DB@DPT_1"
  132. - "C.M.RO4_DB@DPT_1"
  133. - "C.M.RO1_DB@DPT_2"
  134. - "C.M.RO2_DB@DPT_2"
  135. - "C.M.RO3_DB@DPT_2"
  136. - "C.M.RO4_DB@DPT_2"