David 1 month ago
parent
commit
a12db39dd9

+ 1 - 5
models_processing/model_tf/tf_bp_pre.py

@@ -71,15 +71,11 @@ def model_prediction_bp():
                 pre_data['howLongAgo'] = int(args['howLongAgo'])
                 res_cols += ['howLongAgo']
         else:
-            pre_data['cdq'] = args.get('cdq', 1)
-            pre_data['dq'] = args.get('dq', 1)
-            pre_data['zq'] = args.get('zq', 1)
             pre_data['power_forecast'] = res[:len(pre_data)]
             pre_data.rename(columns={args['col_time']: 'date_time'}, inplace=True)
-            res_cols = ['date_time', 'power_forecast', 'farm_id', 'cdq', 'dq', 'zq']
+            res_cols = ['date_time', 'power_forecast', 'farm_id']
 
         pre_data = pre_data[res_cols]
-
         pre_data['power_forecast'] = pre_data['power_forecast'].round(2)
         pre_data.loc[pre_data['power_forecast'] > float(args['cap']), 'power_forecast'] = float(args['cap'])
         pre_data.loc[pre_data['power_forecast'] < 0, 'power_forecast'] = 0

+ 1 - 4
models_processing/model_tf/tf_cnn_pre.py

@@ -72,12 +72,9 @@ def model_prediction_bp():
                 pre_data['howLongAgo'] = int(args['howLongAgo'])
                 res_cols += ['howLongAgo']
         else:
-            pre_data['cdq'] = args.get('cdq', 1)
-            pre_data['dq'] = args.get('dq', 1)
-            pre_data['zq'] = args.get('zq', 1)
             pre_data['power_forecast'] = res[:len(pre_data)]
             pre_data.rename(columns={args['col_time']: 'date_time'}, inplace=True)
-            res_cols = ['date_time', 'power_forecast', 'farm_id', 'cdq', 'dq', 'zq']
+            res_cols = ['date_time', 'power_forecast', 'farm_id']
         pre_data = pre_data[res_cols]
 
         pre_data['power_forecast'] = pre_data['power_forecast'].round(2)

+ 1 - 4
models_processing/model_tf/tf_lstm_pre.py

@@ -69,12 +69,9 @@ def model_prediction_bp():
                 pre_data['howLongAgo'] = int(args['howLongAgo'])
                 res_cols += ['howLongAgo']
         else:
-            pre_data['cdq'] = args.get('cdq', 1)
-            pre_data['dq'] = args.get('dq', 1)
-            pre_data['zq'] = args.get('zq', 1)
             pre_data['power_forecast'] = res[:len(pre_data)]
             pre_data.rename(columns={args['col_time']: 'date_time'}, inplace=True)
-            res_cols = ['date_time', 'power_forecast', 'farm_id', 'cdq', 'dq', 'zq']
+            res_cols = ['date_time', 'power_forecast', 'farm_id']
         pre_data = pre_data[res_cols]
 
         pre_data['power_forecast'] = pre_data['power_forecast'].round(2)

+ 1 - 4
models_processing/model_tf/tf_test_pre.py

@@ -71,12 +71,9 @@ def model_prediction_test():
                 pre_data['howLongAgo'] = int(args['howLongAgo'])
                 res_cols += ['howLongAgo']
         else:
-            pre_data['cdq'] = args.get('cdq', 1)
-            pre_data['dq'] = args.get('dq', 1)
-            pre_data['zq'] = args.get('zq', 1)
             pre_data['power_forecast'] = res[:len(pre_data)]
             pre_data.rename(columns={args['col_time']: 'date_time'}, inplace=True)
-            res_cols = ['date_time', 'power_forecast', 'farm_id', 'cdq', 'dq', 'zq']
+            res_cols = ['date_time', 'power_forecast', 'farm_id']
         pre_data = pre_data[res_cols]
         pre_data['power_forecast'] = pre_data['power_forecast'].round(2)
         pre_data.loc[pre_data['power_forecast'] > float(args['cap']), 'power_forecast'] = float(args['cap'])