config.yaml 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. # anzhen/config.yaml
  2. project:
  3. plant_name: "anzhen"
  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: 17
  14. train_start_date: "2024-10-09"
  15. train_end_date: "2025-03-24"
  16. val_start_date: "2024-10-09"
  17. val_end_date: "2025-03-24"
  18. test_start_date: "2024-10-09"
  19. test_end_date: "2025-03-24"
  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. - "AR.1#UF_JSFLOW_O"
  43. - "AR.2#UF_JSFLOW_O"
  44. - "AR.1#RO_JSFLOW_O"
  45. - "AR.2#RO_JSFLOW_O"
  46. - "AR.1#UF_JSPRESS_O"
  47. - "AR.2#UF_JSPRESS_O"
  48. - "AR.1#RO_JSPRESS_O"
  49. - "AR.2#RO_JSPRESS_O"
  50. - "AR.1#RO_EDJSPRESS_O"
  51. - "AR.1#RO_SDJSPRESS_O"
  52. - "AR.2#RO_EDJSPRESS_O"
  53. - "AR.2#RO_SDJSPRESS_O"
  54. - "AR.ZJS_TEMP_O"
  55. - "AR.ZJS_ZD_O"
  56. - "AR.RO_JSDD_O"
  57. - "AR.RO_JSORP_O"
  58. - "AR.RO_JSPH_O"
  59. - "AR.1#UF_V_FB_O"
  60. - "AR.2#UF_V_FB_O"
  61. - "AR.1#UFBWB_FRE_FB_O"
  62. - "AR.2#UFBWB_FRE_FB_O"
  63. - "AR.1#RODJB_FRE_FB_O"
  64. - "AR.1#ROGYB_FRE_FB_O"
  65. - "AR.1#RODJB_CZ_O"
  66. - "AR.1#ROGYB_CZ_O"
  67. - "AR.2#RODJB_CZ_O"
  68. - "AR.2#ROGYB_CZ_O"
  69. - "AR.ROGSB_FRE_FB_O"
  70. - "AR.UFGSB_FRE_FB_O"
  71. - "AR.V_UF1_TJV_KD_FB"
  72. - "AR.V_UF2_TJV_KD_FB"
  73. - "AR.CS_LEVEL_O"
  74. - "AR.UF_CSLEVEL_O"
  75. - "AR.UF1_SSD_KMYC"
  76. - "AR.UF2_SSD_KMYC"
  77. - "AR.RO1_2D_YC"
  78. - "AR.PUBLIC_BY_REAL_1"
  79. - "1#RO_CSFLOW"
  80. # 最终预测目标列
  81. target_columns:
  82. - "AR.UF1_SSD_KMYC"
  83. - "AR.UF2_SSD_KMYC"
  84. - "AR.RO1_2D_YC"
  85. - "AR.PUBLIC_BY_REAL_1"
  86. - "1#RO_CSFLOW"