David 2 months ago
parent
commit
90d97ddab8
1 changed files with 2 additions and 3 deletions
  1. 2 3
      models_processing/model_koi/tf_bp_pre.py

+ 2 - 3
models_processing/model_koi/tf_bp_pre.py

@@ -49,9 +49,8 @@ def model_prediction_bp():
         feature_scaler, target_scaler = get_scaler_model_from_mongo(args)
         scaled_pre_x = dh.pre_data_handler(pre_data, feature_scaler, opt, bp_data=True)
         bp.get_model(args)
-        # result = bp.predict(scaled_pre_x, args)
-        result = list(chain.from_iterable(target_scaler.inverse_transform([bp.predict(scaled_pre_x).flatten()])))
-        pre_data['power_forecast'] = result[:len(pre_data)]
+        res = list(chain.from_iterable(target_scaler.inverse_transform([bp.predict(scaled_pre_x).flatten()])))
+        pre_data['power_forecast'] = res[:len(pre_data)]
         pre_data['farm_id'] = 'J00083'
         pre_data['cdq'] = 1
         pre_data['dq'] = 1