From: Jérôme Benoit Date: Fri, 1 Aug 2025 00:06:38 +0000 (+0200) Subject: fix(qav3): revert thresholding logic change X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=678f5dbbdddf161f46f18f5e331f1e0dd3b3a1b7;p=freqai-strategies.git fix(qav3): revert thresholding logic change Signed-off-by: Jérôme Benoit --- diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py index 6098d3a..a6784cd 100644 --- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py +++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py @@ -558,10 +558,13 @@ class QuickAdapterRegressorV3(BaseRegressionModel): @staticmethod def get_pred_min_max(pred_extrema: pd.Series) -> tuple[pd.Series, pd.Series]: - minima_indices = sp.signal.find_peaks(-pred_extrema)[0] - maxima_indices = sp.signal.find_peaks(pred_extrema)[0] + n_pred_minima = max(1, sp.signal.find_peaks(-pred_extrema)[0].size) + n_pred_maxima = max(1, sp.signal.find_peaks(pred_extrema)[0].size) - return pred_extrema.iloc[minima_indices], pred_extrema.iloc[maxima_indices] + sorted_pred_extrema = pred_extrema.sort_values(ascending=True) + return sorted_pred_extrema.iloc[:n_pred_minima], sorted_pred_extrema.iloc[ + -n_pred_maxima: + ] @staticmethod def soft_extremum_min_max(