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

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: 24
  train_start_date: "2024-10-08"
  train_end_date: "2026-02-13"
  val_start_date: "2024-10-08"
  val_end_date: "2026-02-13"
  test_start_date: "2024-10-08"
  test_end_date: "2026-02-13"

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_out"
    - "ns=3;s=AI_ROJSLL_OUT"
    - "ns=3;s=AI_UFCSLL_OUT"
    - "ns=3;s=RO_1DJSLL_SSD"
    - "ns=3;s=RO_2DJSLL_SSD"
    - "ns=3;s=RO_NS_SSD"
    - "ns=3;s=AI_JYCSLL1_OUT"
    - "ns=3;s=AI_RODJYL_OUT"
    - "ns=3;s=AI_ROJSYL_OUT"
    - "ns=3;s=AI_UFCSYL_OUT"
    - "ns=3;s=AI_JYCIPPH_OUT"
    - "ns=3;s=AI_JYCSDD_OUT"
    - "ns=3;s=AI_UFCSZD_OUT"
    - "ns=3;s=AI_ROCSDD_OUT"
    - "ns=3;s=AI_UFJSORP_OUT"
    - "ns=3;s=AI_UFJSPH_OUT"
    - "ns=3;s=AI_UFJSYW_OUT"
    - "ns=3;s=AI_JYROCSYW_OUT"
    - "ns=3;s=AI_JYSYW_OUT"
    - "ns=3;s=AI_RODJB_FR_OUT"
    - "ns=3;s=AI_ROGSB_FR_OUT"
    - "ns=3;s=AI_ROGYB_FR_OUT"
    - "ns=3;s=AI_UFFXB_FR_OUT"
    - "ns=3;s=AI_UFCSB_FR_OUT"
    - "ns=3;s=UF_TMP"
    - "ns=3;s=RO_CHA1YL_SSD"
    - "ns=3;s=RO_CHA2YL_SSD"
    - "ns=3;s=RO_ZCS_SSD"
  
  # 最终预测目标列
  target_columns:
    - "ns=3;s=UF_TMP"
    - "ns=3;s=RO_CHA1YL_SSD"
    - "ns=3;s=RO_CHA2YL_SSD"
    - "ns=3;s=RO_ZCS_SSD"