]> Piment Noir Git Repositories - freqai-strategies.git/commitdiff
fix(qav3): kmeans MO selection clause guard on the number of samples
authorJérôme Benoit <jerome.benoit@piment-noir.org>
Thu, 19 Jun 2025 09:25:38 +0000 (11:25 +0200)
committerJérôme Benoit <jerome.benoit@piment-noir.org>
Thu, 19 Jun 2025 09:25:38 +0000 (11:25 +0200)
Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py

index 157d616d4e612b59118112bacafe91a5595933e7..c615b8c8878dcf6e090b27d680b136ee4580d937 100644 (file)
@@ -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(