From 7e2f857aa5fcc2273d4d114ff9d0e65ddbcac741 Mon Sep 17 00:00:00 2001 From: Robert Caulk Date: Sun, 18 Jun 2023 11:30:33 +0200 Subject: [PATCH] Update BasePyTorchClassifier.py --- freqtrade/freqai/base_models/BasePyTorchClassifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/freqtrade/freqai/base_models/BasePyTorchClassifier.py b/freqtrade/freqai/base_models/BasePyTorchClassifier.py index 8a4e15308..c47c5069a 100644 --- a/freqtrade/freqai/base_models/BasePyTorchClassifier.py +++ b/freqtrade/freqai/base_models/BasePyTorchClassifier.py @@ -91,7 +91,7 @@ class BasePyTorchClassifier(BasePyTorchModel): pred_df = DataFrame(predicted_classes_str, columns=[dk.label_list[0]]) pred_df = pd.concat([pred_df, pred_df_prob], axis=1) - if self.freqai_info.get("DI_threshold", 0) > 0: + if dk.feature_pipeline["di"]: dk.DI_values = dk.feature_pipeline["di"].di_values else: dk.DI_values = np.zeros(len(outliers.index))