# longting/config.yaml
project:
  plant_name: "longting"

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: 10
  train_start_date: "2025-11-28"
  train_end_date: "2026-02-20"
  val_start_date: "2025-11-28"
  val_end_date: "2026-02-20"
  test_start_date: "2025-11-28"
  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"
    - "water_in"
    - "water_out"
    - "RO1_TYL"
    - "RO2_TYL"
    - "UF1Per"
    - "UF2Per"
    - "2#RODJB_Eff"
    - "1#RODJB_Eff"
    - "2#ROGYB_Eff"
    - "1#ROGYB_Eff"
    - "ROHSL"
    - "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=RO1_1DUAN_CS_FLOW"
    - "ns=3;s=ZJS_PRESS_O"
    - "ns=3;s=ZJS_TEMP_O"
    - "ns=3;s=ZJS_ZD_O"
    - "ns=3;s=PUBLIC_RO1_MTL"
    - "ns=3;s=PUBLIC_RO2_MTL"
    - "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"
    - "ns=3;s=PUBLIC_BY_REAL_1"
    - "ns=3;s=PUBLIC_BY_REAL_2"
  
  # 最终预测目标列
  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"
    - "ns=3;s=PUBLIC_BY_REAL_1"
    - "ns=3;s=PUBLIC_BY_REAL_2"