python - 如何在 python 的 sklearn.kmeans(n_clusters,init,....) 中将一些节点作为 init 传递

Question

如何将 2 个节点作为质心传递给 sklearn.kmeans(n_clusters,init,....) 而不是随机或 kmeans++？算法['kmeans'] = KMeans(n_clusters=k_clusters,init='random', n_init=200)

Answer 1

sklearn还允许用户输入形状的质心(n_clusters, n_features)。

文档：https ://scikit-learn.org/stable/modules/generated/sklearn.cluster.KMeans.html

对于小于n_clusters，您可以使用随机初始化并通过它。