config.yaml 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131
  1. # longting/config.yaml
  2. project:
  3. plant_name: "longting"
  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: 10
  14. train_start_date: "2025-11-28"
  15. train_end_date: "2026-02-20"
  16. val_start_date: "2025-11-28"
  17. val_end_date: "2026-02-20"
  18. test_start_date: "2025-11-28"
  19. test_end_date: "2026-02-20"
  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_in"
  43. - "water_out"
  44. - "RO1_TYL"
  45. - "RO2_TYL"
  46. - "UF1Per"
  47. - "UF2Per"
  48. - "2#RODJB_Eff"
  49. - "1#RODJB_Eff"
  50. - "2#ROGYB_Eff"
  51. - "1#ROGYB_Eff"
  52. - "ROHSL"
  53. - "ns=3;s=1#RO_CSDD_O"
  54. - "ns=3;s=1#RO_CSPRESS_O"
  55. - "ns=3;s=1#RO_EDCSFLOW_O"
  56. - "ns=3;s=1#RO_EDJSPRESS_O"
  57. - "ns=3;s=1#RO_EDNSPRESS_O"
  58. - "ns=3;s=1#RO_JSFLOW_O"
  59. - "ns=3;s=1#RO_JSPRESS_O"
  60. - "ns=3;s=1#RO_NSFLOW_O"
  61. - "ns=3;s=1#RO_SDCSFLOW_O"
  62. - "ns=3;s=1#RO_SDJSPRESS_O"
  63. - "ns=3;s=1#RO_SDNSPRESS_O"
  64. - "ns=3;s=1#RODJB_CUR_FB_O"
  65. - "ns=3;s=1#RODJB_CZ_O"
  66. - "ns=3;s=1#RODJB_FRE_FB_O"
  67. - "ns=3;s=1#ROGYB_CUR_FB_O"
  68. - "ns=3;s=1#ROGYB_CZ_O"
  69. - "ns=3;s=1#ROGYB_FRE_FB_O"
  70. - "ns=3;s=1#UF_CSPRESS_O"
  71. - "ns=3;s=1#UF_JSFLOW_O"
  72. - "ns=3;s=1#UF_JSPRESS_O"
  73. - "ns=3;s=1#UF_V_FB_O"
  74. - "ns=3;s=1#UFBWB_CUR_FB_O"
  75. - "ns=3;s=1#UFBWB_FRE_FB_O"
  76. - "ns=3;s=2#RO_CSDD_O"
  77. - "ns=3;s=2#RO_CSPRESS_O"
  78. - "ns=3;s=2#RO_EDCSFLOW_O"
  79. - "ns=3;s=2#RO_EDJSPRESS_O"
  80. - "ns=3;s=2#RO_EDNSPRESS_O"
  81. - "ns=3;s=2#RO_JSFLOW_O"
  82. - "ns=3;s=2#RO_JSPRESS_O"
  83. - "ns=3;s=2#RO_NSFLOW_O"
  84. - "ns=3;s=2#RO_SDCSFLOW_O"
  85. - "ns=3;s=2#RO_SDJSPRESS_O"
  86. - "ns=3;s=2#RO_SDNSPRESS_O"
  87. - "ns=3;s=2#RODJB_CUR_FB_O"
  88. - "ns=3;s=2#RODJB_CZ_O"
  89. - "ns=3;s=2#RODJB_FRE_FB_O"
  90. - "ns=3;s=2#ROGYB_CUR_FB_O"
  91. - "ns=3;s=2#ROGYB_CZ_O"
  92. - "ns=3;s=2#ROGYB_FRE_FB_O"
  93. - "ns=3;s=2#UF_CSPRESS_O"
  94. - "ns=3;s=2#UF_JSFLOW_O"
  95. - "ns=3;s=2#UF_JSPRESS_O"
  96. - "ns=3;s=2#UF_V_FB_O"
  97. - "ns=3;s=2#UFBWB_CUR_FB_O"
  98. - "ns=3;s=2#UFBWB_FRE_FB_O"
  99. - "ns=3;s=RO_JSDD_O"
  100. - "ns=3;s=RO_JSORP_O"
  101. - "ns=3;s=RO_JSPH_O"
  102. - "ns=3;s=RO1_1DUAN_CS_FLOW"
  103. - "ns=3;s=ZJS_PRESS_O"
  104. - "ns=3;s=ZJS_TEMP_O"
  105. - "ns=3;s=ZJS_ZD_O"
  106. - "ns=3;s=PUBLIC_RO1_MTL"
  107. - "ns=3;s=PUBLIC_RO2_MTL"
  108. - "ns=3;s=UF1_SSD_KMYC"
  109. - "ns=3;s=UF2_SSD_KMYC"
  110. - "ns=3;s=RO1_1D_YC"
  111. - "ns=3;s=RO1_2D_YC"
  112. - "ns=3;s=RO2_1D_YC"
  113. - "ns=3;s=RO2_2D_YC"
  114. - "ns=3;s=PUBLIC_BY_REAL_1"
  115. - "ns=3;s=PUBLIC_BY_REAL_2"
  116. # 最终预测目标列
  117. target_columns:
  118. - "ns=3;s=UF1_SSD_KMYC"
  119. - "ns=3;s=UF2_SSD_KMYC"
  120. - "ns=3;s=RO1_1D_YC"
  121. - "ns=3;s=RO1_2D_YC"
  122. - "ns=3;s=RO2_1D_YC"
  123. - "ns=3;s=RO2_2D_YC"
  124. - "ns=3;s=PUBLIC_BY_REAL_1"
  125. - "ns=3;s=PUBLIC_BY_REAL_2"