From 295ba21389565948268756829949cc7657098afd Mon Sep 17 00:00:00 2001 From: Mark Regan Date: Thu, 20 Oct 2022 12:05:37 +0100 Subject: [PATCH 1/2] Make extra_returns_per_train values available during backtest --- freqtrade/freqai/data_kitchen.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/freqtrade/freqai/data_kitchen.py b/freqtrade/freqai/data_kitchen.py index 52261d341..38fe670ec 100644 --- a/freqtrade/freqai/data_kitchen.py +++ b/freqtrade/freqai/data_kitchen.py @@ -957,6 +957,9 @@ class FreqaiDataKitchen: append_df[f"{label}_mean"] = self.data["labels_mean"][label] append_df[f"{label}_std"] = self.data["labels_std"][label] + for extra_col in self.data["extra_returns_per_train"]: + append_df[f"{extra_col}"] = self.data["extra_returns_per_train"][extra_col] + append_df["do_predict"] = do_predict if self.freqai_config["feature_parameters"].get("DI_threshold", 0) > 0: append_df["DI_values"] = self.DI_values From 073ce1659eff7923bd5468702daecc553b044d6e Mon Sep 17 00:00:00 2001 From: Mark Regan Date: Thu, 20 Oct 2022 14:26:10 +0100 Subject: [PATCH 2/2] remove un-used f-string --- freqtrade/freqai/data_kitchen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/freqtrade/freqai/data_kitchen.py b/freqtrade/freqai/data_kitchen.py index 38fe670ec..39310a15e 100644 --- a/freqtrade/freqai/data_kitchen.py +++ b/freqtrade/freqai/data_kitchen.py @@ -958,7 +958,7 @@ class FreqaiDataKitchen: append_df[f"{label}_std"] = self.data["labels_std"][label] for extra_col in self.data["extra_returns_per_train"]: - append_df[f"{extra_col}"] = self.data["extra_returns_per_train"][extra_col] + append_df["{extra_col}"] = self.data["extra_returns_per_train"][extra_col] append_df["do_predict"] = do_predict if self.freqai_config["feature_parameters"].get("DI_threshold", 0) > 0: