]> Piment Noir Git Repositories - freqai-strategies.git/commitdiff
refactor(qav3): add tunable to extrema smoothing window
authorJérôme Benoit <jerome.benoit@piment-noir.org>
Thu, 20 Mar 2025 14:51:56 +0000 (15:51 +0100)
committerJérôme Benoit <jerome.benoit@piment-noir.org>
Thu, 20 Mar 2025 14:51:56 +0000 (15:51 +0100)
Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
quickadapter/user_data/strategies/QuickAdapterV3.py

index 37e63b7785c21ba5f428076daeea364070d9bf14..22c5f36d405918cea1da81cb7b5f597b3a6e0d13 100644 (file)
@@ -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: