We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 56af60d commit 1b7a172Copy full SHA for 1b7a172
autosklearn/metalearning/metalearning/clustering/gmeans.py
@@ -69,7 +69,7 @@ def fit(self, X):
69
break
70
71
# Refinement
72
- KMeans = sklearn.cluster.KMeans(n_clusters=1, n_init=1,
+ KMeans = sklearn.cluster.KMeans(n_clusters=len(cluster_centers), n_init=1,
73
init=np.array(cluster_centers),
74
random_state=self.random_state)
75
KMeans.fit(X)
0 commit comments