From 678f5dbbdddf161f46f18f5e331f1e0dd3b3a1b7 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Fri, 1 Aug 2025 02:06:38 +0200 Subject: [PATCH] fix(qav3): revert thresholding logic change MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/QuickAdapterRegressorV3.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) 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( -- 2.43.0