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

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: 17
  train_start_date: "2024-10-09"
  train_end_date: "2025-03-24"
  val_start_date: "2024-10-09"
  val_end_date: "2025-03-24"
  test_start_date: "2024-10-09"
  test_end_date: "2025-03-24"

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"
    - "AR.1#UF_JSFLOW_O"
    - "AR.2#UF_JSFLOW_O"
    - "AR.1#RO_JSFLOW_O"
    - "AR.2#RO_JSFLOW_O"
    - "AR.1#UF_JSPRESS_O"
    - "AR.2#UF_JSPRESS_O"
    - "AR.1#RO_JSPRESS_O"
    - "AR.2#RO_JSPRESS_O"
    - "AR.1#RO_EDJSPRESS_O"
    - "AR.1#RO_SDJSPRESS_O"
    - "AR.2#RO_EDJSPRESS_O"
    - "AR.2#RO_SDJSPRESS_O"
    - "AR.ZJS_TEMP_O"
    - "AR.ZJS_ZD_O"
    - "AR.RO_JSDD_O"
    - "AR.RO_JSORP_O"
    - "AR.RO_JSPH_O"
    - "AR.1#UF_V_FB_O"
    - "AR.2#UF_V_FB_O"
    - "AR.1#UFBWB_FRE_FB_O"
    - "AR.2#UFBWB_FRE_FB_O"
    - "AR.1#RODJB_FRE_FB_O"
    - "AR.1#ROGYB_FRE_FB_O"
    - "AR.1#RODJB_CZ_O"
    - "AR.1#ROGYB_CZ_O"
    - "AR.2#RODJB_CZ_O"
    - "AR.2#ROGYB_CZ_O"
    - "AR.ROGSB_FRE_FB_O"
    - "AR.UFGSB_FRE_FB_O"
    - "AR.V_UF1_TJV_KD_FB"
    - "AR.V_UF2_TJV_KD_FB"
    - "AR.CS_LEVEL_O"
    - "AR.UF_CSLEVEL_O"
    - "AR.UF1_SSD_KMYC"
    - "AR.UF2_SSD_KMYC"
    - "AR.RO1_2D_YC"
    - "AR.PUBLIC_BY_REAL_1"
    - "1#RO_CSFLOW"
  
  # 最终预测目标列
  target_columns:
    - "AR.UF1_SSD_KMYC"
    - "AR.UF2_SSD_KMYC"
    - "AR.RO1_2D_YC"
    - "AR.PUBLIC_BY_REAL_1"
    - "1#RO_CSFLOW"