Explorar o código

1.盐城预测模型更新

zhanghao hai 2 días
pai
achega
bd5224586d

+ 7 - 5
models/prediction_models/20min/yancheng/config.yaml

@@ -3,7 +3,7 @@ project:
   plant_name: "yancheng"
 
 files:
-  dataset_dir: "datasets"
+  dataset_dir: "datasets_yancheng"
   file_pattern: "data_process_{}.csv"
   model_filename: "model.pth"
   scaler_filename: "scaler.pkl"
@@ -12,19 +12,21 @@ files:
 
 data_split:
   start_files: 1
-  end_files: 4
+  end_files: 5
   train_start_date: "2026-01-23"
-  train_end_date: "2026-02-20"
+  train_end_date: "2026-05-14"
   val_start_date: "2026-01-23"
-  val_end_date: "2026-02-20"
+  val_end_date: "2026-05-14"
   test_start_date: "2026-01-23"
-  test_end_date: "2026-02-20"
+  test_end_date: "2026-05-14"
 
 model_params:
   seq_len: 10
   output_size: 5
   step_size: 5
   resolution: 60
+  feature_num: 68
+  labels_num: 6
   hidden_size: 64
   num_layers: 1
   dropout: 0.0

BIN=BIN
models/prediction_models/20min/yancheng/model.pth


BIN=BIN
models/prediction_models/20min/yancheng/scaler.pkl