CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-12-19 06:11:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update BasePyTorchRegressor.py

Browse Source 
Denormalization of prediction added to te PytorchMLP Model
This commit is contained in:
vinistation
2023-04-28 14:48:16 -05:00
committed by GitHub
parent 2a9e50a6a9
commit d1eb6d4fed
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
freqtrade/freqai/base_models/BasePyTorchRegressor.py
View File

@@ -47,4 +47,5 @@ class BasePyTorchRegressor(BasePyTorchModel):
y = self.model.model(x) y = self.model.model(x)
y = y.cpu() y = y.cpu()
pred_df = DataFrame(y.detach().numpy(), columns=[dk.label_list[0]]) 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) return (pred_df, dk.do_predict)