|
|
@@ -3,7 +3,7 @@ project:
|
|
|
plant_name: "jianding"
|
|
|
|
|
|
files:
|
|
|
- dataset_dir: "datasets"
|
|
|
+ dataset_dir: "datasets_jianding"
|
|
|
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: 24
|
|
|
+ end_files: 26
|
|
|
train_start_date: "2024-10-08"
|
|
|
- train_end_date: "2026-02-13"
|
|
|
+ train_end_date: "2026-05-14"
|
|
|
val_start_date: "2024-10-08"
|
|
|
- val_end_date: "2026-02-13"
|
|
|
+ val_end_date: "2026-05-14"
|
|
|
test_start_date: "2024-10-08"
|
|
|
- test_end_date: "2026-02-13"
|
|
|
+ test_end_date: "2026-05-14"
|
|
|
|
|
|
model_params:
|
|
|
seq_len: 10
|
|
|
output_size: 5
|
|
|
step_size: 5
|
|
|
resolution: 60
|
|
|
+ feature_num: 32
|
|
|
+ labels_num: 4
|
|
|
hidden_size: 64
|
|
|
num_layers: 1
|
|
|
dropout: 0.0
|
|
|
@@ -37,7 +39,7 @@ training_params:
|
|
|
scheduler_gamma: 0.9
|
|
|
patience: 200
|
|
|
min_delta: 1.0e-10
|
|
|
- device: 1 # 对应 args.device
|
|
|
+ device: 1
|
|
|
random_seed: 1314
|
|
|
|
|
|
sensors:
|
