freqai_info -> ft_params

This commit is contained in:
Robert Caulk 2024-09-26 17:54:14 +02:00 committed by GitHub
parent 123909cdac
commit d18d8cf0ea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -141,7 +141,7 @@ class PyTorchTransformerRegressor(BasePyTorchRegressor):
pred_df = pd.DataFrame(yb.detach().numpy(), columns=dk.label_list)
pred_df, _, _ = dk.label_pipeline.inverse_transform(pred_df)
if self.freqai_info.get("DI_threshold", 0) > 0:
if self.ft_params.get("DI_threshold", 0) > 0:
dk.DI_values = dk.feature_pipeline["di"].di_values
else:
dk.DI_values = np.zeros(outliers.shape[0])