|
|
@@ -64,7 +64,7 @@ def model_prediction_lstm():
|
|
|
ts.get_model(args)
|
|
|
dh.opt.features = json.loads(ts.model_params)['Model']['features'].split(',')
|
|
|
scaled_pre_x, pre_data = dh.pre_data_handler(pre_data, feature_scaler, time_series=args['time_series'])
|
|
|
- res = list(chain.from_iterable(target_scaler.inverse_transform(ts.predict(scaled_pre_x))))
|
|
|
+ res = list(chain.from_iterable(target_scaler.inverse_transform(ts.predict(scaled_pre_x)[1])))
|
|
|
pre_data['farm_id'] = args.get('farm_id', 'null')
|
|
|
if int(args.get('algorithm_test', 0)):
|
|
|
pre_data[args['model_name']] = res[:len(pre_data)]
|
David