Add size check before trying to split for GMeans (#732)

* Add size check before trying to split * Rebase to new code Co-authored-by: chico <[email protected]>
automl · Feb 24, 2021 · d3aa95e · d3aa95e
1 parent b66d2c7
commit d3aa95e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/autosklearn/metalearning/metalearning/clustering/gmeans.py b/autosklearn/metalearning/metalearning/clustering/gmeans.py
@@ -34,6 +34,10 @@ def fit(self, X):
                     indices = KMeans.labels_ == i
                     X_ = X[indices]
 
+                    if np.sum(indices) < self.minimum_samples_per_cluster*2:
+                        cluster_centers.append(cluster_center)
+                        continue
+
                     for i in range(10):
                         KMeans_ = sklearn.cluster.KMeans(n_clusters=2,
                                                          n_init=self.n_init,