From 1ebb952cb75a739965c911c89f5546e5dde2cc95 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Thu, 19 Jun 2025 11:25:38 +0200 Subject: [PATCH] fix(qav3): kmeans MO selection clause guard on the number of samples MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py index 157d616..c615b8c 100644 --- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py +++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py @@ -666,7 +666,7 @@ class QuickAdapterRegressorV3(BaseRegressionModel): return np.sum(np_weights * (ideal_point - normalized_matrix), axis=1) elif metric in {"kmeans", "kmeans2"}: if n_samples < 2: - return np.ndarray([]) + return np.full(n_samples, np.inf) n_clusters = min(max(2, int(np.sqrt(n_samples / 2))), 10, n_samples) if metric == "kmeans": kmeans = sklearn.cluster.KMeans( -- 2.43.0