| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119 |
- # yancheng/config.yaml
- project:
- plant_name: "yancheng"
- files:
- dataset_dir: "datasets"
- file_pattern: "data_process_{}.csv"
- model_filename: "model.pth"
- scaler_filename: "scaler.pkl"
- output_csv_filename: "predictions.csv"
- edge_index_filename: "edge_index.pt"
- data_split:
- start_files: 1
- end_files: 4
- train_start_date: "2026-01-23"
- train_end_date: "2026-02-20"
- val_start_date: "2026-01-23"
- val_end_date: "2026-02-20"
- test_start_date: "2026-01-23"
- test_end_date: "2026-02-20"
- model_params:
- seq_len: 10
- output_size: 5
- step_size: 5
- resolution: 60
- hidden_size: 64
- num_layers: 1
- dropout: 0.0
- training_params:
- epochs: 200
- lr: 0.01
- batch_size: 512
- scheduler_step_size: 100
- scheduler_gamma: 0.9
- patience: 200
- min_delta: 1.0e-10
- device: 1 # 对应 args.device
- random_seed: 1314
- sensors:
- # 输入传感器列
- required_columns:
- - "index"
- - "ns=3;s=1#RO_CSDD_O"
- - "ns=3;s=1#RO_CSPRESS_O"
- - "ns=3;s=1#RO_EDCSFLOW_O"
- - "ns=3;s=1#RO_EDJSPRESS_O"
- - "ns=3;s=1#RO_EDNSPRESS_O"
- - "ns=3;s=1#RO_JSFLOW_O"
- - "ns=3;s=1#RO_JSPRESS_O"
- - "ns=3;s=1#RO_NSFLOW_O"
- - "ns=3;s=1#RO_SDCSFLOW_O"
- - "ns=3;s=1#RO_SDJSPRESS_O"
- - "ns=3;s=1#RO_SDNSPRESS_O"
- - "ns=3;s=1#RODJB_CUR_FB_O"
- - "ns=3;s=1#RODJB_CZ_O"
- - "ns=3;s=1#RODJB_FRE_FB_O"
- - "ns=3;s=1#ROGYB_CUR_FB_O"
- - "ns=3;s=1#ROGYB_CZ_O"
- - "ns=3;s=1#ROGYB_FRE_FB_O"
- - "ns=3;s=1#UF_CSPRESS_O"
- - "ns=3;s=1#UF_JSFLOW_O"
- - "ns=3;s=1#UF_JSPRESS_O"
- - "ns=3;s=1#UF_V_FB_O"
- - "ns=3;s=1#UFBWB_CUR_FB_O"
- - "ns=3;s=1#UFBWB_FRE_FB_O"
- - "ns=3;s=2#RO_CSDD_O"
- - "ns=3;s=2#RO_CSPRESS_O"
- - "ns=3;s=2#RO_EDCSFLOW_O"
- - "ns=3;s=2#RO_EDJSPRESS_O"
- - "ns=3;s=2#RO_EDNSPRESS_O"
- - "ns=3;s=2#RO_JSFLOW_O"
- - "ns=3;s=2#RO_JSPRESS_O"
- - "ns=3;s=2#RO_NSFLOW_O"
- - "ns=3;s=2#RO_SDCSFLOW_O"
- - "ns=3;s=2#RO_SDJSPRESS_O"
- - "ns=3;s=2#RO_SDNSPRESS_O"
- - "ns=3;s=2#RODJB_CUR_FB_O"
- - "ns=3;s=2#RODJB_CZ_O"
- - "ns=3;s=2#RODJB_FRE_FB_O"
- - "ns=3;s=2#ROGYB_CUR_FB_O"
- - "ns=3;s=2#ROGYB_CZ_O"
- - "ns=3;s=2#ROGYB_FRE_FB_O"
- - "ns=3;s=2#UF_CSPRESS_O"
- - "ns=3;s=2#UF_JSFLOW_O"
- - "ns=3;s=2#UF_JSPRESS_O"
- - "ns=3;s=2#UF_V_FB_O"
- - "ns=3;s=2#UFBWB_CUR_FB_O"
- - "ns=3;s=2#UFBWB_FRE_FB_O"
- - "ns=3;s=RO_JSDD_O"
- - "ns=3;s=RO_JSORP_O"
- - "ns=3;s=RO_JSPH_O"
- - "ns=3;s=RO_WSDD_O"
- - "ns=3;s=ZJS_ZD_O"
- - "water_out"
- - "water_in"
- - "ns=3;s=V_UF1_TJV_KD_FB"
- - "ns=3;s=V_UF2_TJV_KD_FB"
- - "ns=3;s=ZJS_PRESS_O"
- - "ns=3;s=ZJS_TEMP_O"
- - "ns=3;s=UF_CS_ZD_O"
- - "ns=3;s=UF1_SSD_KMYC"
- - "ns=3;s=UF2_SSD_KMYC"
- - "ns=3;s=RO1_1D_YC"
- - "ns=3;s=RO1_2D_YC"
- - "ns=3;s=RO2_1D_YC"
- - "ns=3;s=RO2_2D_YC"
-
- # 最终预测目标列
- target_columns:
- - "ns=3;s=UF1_SSD_KMYC"
- - "ns=3;s=UF2_SSD_KMYC"
- - "ns=3;s=RO1_1D_YC"
- - "ns=3;s=RO1_2D_YC"
- - "ns=3;s=RO2_1D_YC"
- - "ns=3;s=RO2_2D_YC"
|
