From 3cd0a417726b818d3735617abffae14a45916317 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Thu, 20 Mar 2025 15:51:56 +0100 Subject: [PATCH] refactor(qav3): add tunable to extrema smoothing window MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- quickadapter/user_data/strategies/QuickAdapterV3.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/quickadapter/user_data/strategies/QuickAdapterV3.py b/quickadapter/user_data/strategies/QuickAdapterV3.py index 37e63b7..22c5f36 100644 --- a/quickadapter/user_data/strategies/QuickAdapterV3.py +++ b/quickadapter/user_data/strategies/QuickAdapterV3.py @@ -331,7 +331,10 @@ class QuickAdapterV3(IStrategy): dataframe.at[mp, EXTREMA_COLUMN] = 1 dataframe["minima"] = np.where(dataframe[EXTREMA_COLUMN] == -1, -1, 0) dataframe["maxima"] = np.where(dataframe[EXTREMA_COLUMN] == 1, 1, 0) - dataframe[EXTREMA_COLUMN] = self.smooth_extrema(dataframe[EXTREMA_COLUMN], 5) + dataframe[EXTREMA_COLUMN] = self.smooth_extrema( + dataframe[EXTREMA_COLUMN], + self.freqai_info.get("extrema_smoothing_window", 5), + ) return dataframe def populate_indicators(self, dataframe: DataFrame, metadata: dict) -> DataFrame: -- 2.43.0