From d1eb6d4fed2a6a8e5d01c457f6ea529e7ef29208 Mon Sep 17 00:00:00 2001 From: vinistation <60822570+vinistation@users.noreply.github.com> Date: Fri, 28 Apr 2023 14:48:16 -0500 Subject: [PATCH] Update BasePyTorchRegressor.py Denormalization of prediction added to te PytorchMLP Model --- freqtrade/freqai/base_models/BasePyTorchRegressor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/freqtrade/freqai/base_models/BasePyTorchRegressor.py b/freqtrade/freqai/base_models/BasePyTorchRegressor.py index ea6fabe49..bc47731b7 100644 --- a/freqtrade/freqai/base_models/BasePyTorchRegressor.py +++ b/freqtrade/freqai/base_models/BasePyTorchRegressor.py @@ -47,4 +47,5 @@ class BasePyTorchRegressor(BasePyTorchModel): y = self.model.model(x) y = y.cpu() pred_df = DataFrame(y.detach().numpy(), columns=[dk.label_list[0]]) + pred_df = dk.denormalize_labels_from_metadata(pred_df) return (pred_df, dk.do_predict)