mirror of
https://github.com/freqtrade/freqtrade.git
synced 2025-12-03 18:43:04 +00:00
Update BasePyTorchRegressor.py
Denormalization of prediction added to te PytorchMLP Model
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user