mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-10 10:21:59 +00:00
Update BasePyTorchRegressor.py
Denormalization of prediction added to te PytorchMLP Model
This commit is contained in:
parent
2a9e50a6a9
commit
d1eb6d4fed
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue
Block a user