ソースを参照

增加代码逻辑,避免Nan值

zhanghao 5 ヶ月 前
コミット
ccba063c71
1 ファイル変更1 行追加7 行削除
  1. 1 7
      models/pressure-predictor/gat-lstm_model/20min/predict.py

+ 1 - 7
models/pressure-predictor/gat-lstm_model/20min/predict.py

@@ -581,13 +581,7 @@ class Predictor:
             time_interval = timedelta(minutes=(4 * self.resolution / 60))
             timestamps = [start_time + i * time_interval for i in range(len(predictions))]
 
-            base_columns = [
-                'C.M.UF1_DB@press_PV', 'C.M.UF2_DB@press_PV', 'C.M.UF3_DB@press_PV', 'C.M.UF4_DB@press_PV',
-                'UF1Per','UF2Per','UF3Per','UF4Per',
-                'C.M.RO1_DB@DPT_1', 'C.M.RO2_DB@DPT_1', 'C.M.RO3_DB@DPT_1', 'C.M.RO4_DB@DPT_1',
-                'C.M.RO1_DB@DPT_2', 'C.M.RO2_DB@DPT_2', 'C.M.RO3_DB@DPT_2', 'C.M.RO4_DB@DPT_2',
-            ]
-
+            base_columns = self.target_columns
             pred_columns = [f'{col}_Predicted' for col in base_columns]
             df_result = pd.DataFrame(predictions, columns=pred_columns)
             df_result.insert(0, 'index', timestamps)