03111016

models_processing/model_koi/bp.yaml

@@ -23,7 +23,7 @@ Model:
   valid_data_rate: 0.15
 region: south
 calculate: []
-cap: 50.0
+cap: 153.0
 dataloc: ./data
 env_columns:
 - C_TIME

models_processing/model_koi/cnn.yaml

@@ -23,7 +23,7 @@ Model:
   valid_data_rate: 0.15
 region: south
 calculate: []
-cap: 50.0
+cap: 153.0
 dataloc: ./data
 env_columns:
 - C_TIME

models_processing/model_koi/lstm.yaml

@@ -23,7 +23,7 @@ Model:
   valid_data_rate: 0.15
 region: south
 calculate: []
-cap: 50.0
+cap: 153.0
 dataloc: ./data
 env_columns:
 - C_TIME

models_processing/model_koi/tf_lstm_pre.py

@@ -62,7 +62,7 @@ def model_prediction_bp():
         if args.get('algorithm_test', 0):
             pre_data['model'] = 'lstm'
             pre_data.rename(columns={args['col_time']: 'dateTime'}, inplace=True)
-            res_cols = ['dateTime', 'power_forecast', 'farm_id', 'cdq', 'dq', 'zq', args['target'], 'model']
+            res_cols = ['dateTime', 'power_forecast', 'farm_id', args['target'], 'model']
         else:
             pre_data['cdq'] = args.get('cdq', 1)
             pre_data['dq'] = args.get('dq', 1)