config.yaml 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. # anzhen/config.yaml
  2. project:
  3. plant_name: "anzhen"
  4. files:
  5. dataset_dir: "datasets_anzhen"
  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: 30
  14. train_start_date: "2024-10-09"
  15. train_end_date: "2026-05-14"
  16. val_start_date: "2024-10-09"
  17. val_end_date: "2026-05-14"
  18. test_start_date: "2024-10-09"
  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: 42
  26. labels_num: 4
  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: 512
  34. scheduler_step_size: 100
  35. scheduler_gamma: 0.9
  36. patience: 200
  37. min_delta: 1.0e-10
  38. device: 1
  39. random_seed: 1314
  40. sensors:
  41. # 输入传感器列
  42. required_columns:
  43. - "index"
  44. - "AR.1#UF_JSFLOW_O"
  45. - "AR.2#UF_JSFLOW_O"
  46. - "AR.1#RO_JSFLOW_O"
  47. - "AR.2#RO_JSFLOW_O"
  48. - "AR.1#UF_JSPRESS_O"
  49. - "AR.2#UF_JSPRESS_O"
  50. - "AR.1#RO_JSPRESS_O"
  51. - "AR.2#RO_JSPRESS_O"
  52. - "AR.1#RO_EDJSPRESS_O"
  53. - "AR.1#RO_SDJSPRESS_O"
  54. - "AR.2#RO_EDJSPRESS_O"
  55. - "AR.2#RO_SDJSPRESS_O"
  56. - "AR.ZJS_TEMP_O"
  57. - "AR.ZJS_ZD_O"
  58. - "AR.RO_JSDD_O"
  59. - "AR.RO_JSORP_O"
  60. - "AR.RO_JSPH_O"
  61. - "AR.1#UF_V_FB_O"
  62. - "AR.2#UF_V_FB_O"
  63. - "AR.1#UFBWB_FRE_FB_O"
  64. - "AR.2#UFBWB_FRE_FB_O"
  65. - "AR.1#RODJB_FRE_FB_O"
  66. - "AR.1#ROGYB_FRE_FB_O"
  67. - "AR.1#RODJB_CZ_O"
  68. - "AR.1#ROGYB_CZ_O"
  69. - "AR.2#RODJB_CZ_O"
  70. - "AR.2#ROGYB_CZ_O"
  71. - "AR.ROGSB_FRE_FB_O"
  72. - "AR.UFGSB_FRE_FB_O"
  73. - "AR.V_UF1_TJV_KD_FB"
  74. - "AR.V_UF2_TJV_KD_FB"
  75. - "AR.CS_LEVEL_O"
  76. - "AR.UF_CSLEVEL_O"
  77. - "AR.UF1_SSD_KMYC"
  78. - "AR.UF2_SSD_KMYC"
  79. - "AR.RO1_2D_YC"
  80. - "AR.PUBLIC_BY_REAL_1"
  81. - "1#RO_CSFLOW"
  82. # 最终预测目标列
  83. target_columns:
  84. - "AR.UF1_SSD_KMYC"
  85. - "AR.UF2_SSD_KMYC"
  86. - "AR.RO1_2D_YC"
  87. - "AR.PUBLIC_BY_REAL_1"
  88. - "1#RO_CSFLOW"