|
@@ -82,7 +82,7 @@ def model_prediction_test():
|
|
|
pre_data.rename(columns={args['col_time']: 'date_time'}, inplace=True)
|
|
|
res_cols = ['date_time', 'power_forecast', 'farm_id']
|
|
|
pre_data = pre_data[res_cols]
|
|
|
- pre_data.loc[:, 'power_forecast'] = pre_data['power_forecast'].round(2)
|
|
|
+ pre_data.loc[:, 'power_forecast'] = pre_data.loc[:, 'power_forecast'].apply(lambda x: float(f"{x:.2f}"))
|
|
|
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
|
|
|
insert_data_into_mongo(pre_data, args)
|