03101700

models_processing/model_koi/tf_bp_pre.py

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

models_processing/model_koi/tf_cnn_pre.py

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

models_processing/model_koi/tf_lstm_pre.py

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