config.yaml 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. # jianding/config.yaml
  2. project:
  3. plant_name: "jianding"
  4. files:
  5. dataset_dir: "datasets_jianding"
  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: 26
  14. train_start_date: "2024-10-08"
  15. train_end_date: "2026-05-14"
  16. val_start_date: "2024-10-08"
  17. val_end_date: "2026-05-14"
  18. test_start_date: "2024-10-08"
  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: 32
  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. - "water_out"
  45. - "ns=3;s=AI_ROJSLL_OUT"
  46. - "ns=3;s=AI_UFCSLL_OUT"
  47. - "ns=3;s=RO_1DJSLL_SSD"
  48. - "ns=3;s=RO_2DJSLL_SSD"
  49. - "ns=3;s=RO_NS_SSD"
  50. - "ns=3;s=AI_JYCSLL1_OUT"
  51. - "ns=3;s=AI_RODJYL_OUT"
  52. - "ns=3;s=AI_ROJSYL_OUT"
  53. - "ns=3;s=AI_UFCSYL_OUT"
  54. - "ns=3;s=AI_JYCIPPH_OUT"
  55. - "ns=3;s=AI_JYCSDD_OUT"
  56. - "ns=3;s=AI_UFCSZD_OUT"
  57. - "ns=3;s=AI_ROCSDD_OUT"
  58. - "ns=3;s=AI_UFJSORP_OUT"
  59. - "ns=3;s=AI_UFJSPH_OUT"
  60. - "ns=3;s=AI_UFJSYW_OUT"
  61. - "ns=3;s=AI_JYROCSYW_OUT"
  62. - "ns=3;s=AI_JYSYW_OUT"
  63. - "ns=3;s=AI_RODJB_FR_OUT"
  64. - "ns=3;s=AI_ROGSB_FR_OUT"
  65. - "ns=3;s=AI_ROGYB_FR_OUT"
  66. - "ns=3;s=AI_UFFXB_FR_OUT"
  67. - "ns=3;s=AI_UFCSB_FR_OUT"
  68. - "ns=3;s=UF_TMP"
  69. - "ns=3;s=RO_CHA1YL_SSD"
  70. - "ns=3;s=RO_CHA2YL_SSD"
  71. - "ns=3;s=RO_ZCS_SSD"
  72. # 最终预测目标列
  73. target_columns:
  74. - "ns=3;s=UF_TMP"
  75. - "ns=3;s=RO_CHA1YL_SSD"
  76. - "ns=3;s=RO_CHA2YL_SSD"
  77. - "ns=3;s=RO_ZCS_SSD"