如何在k-means中記錄每次迭代的質心？

Question

通過使用“kmeans.cluster_centers_”，我可以獲得每個集群的最終質心，但是如果我想跟蹤所有迭代中的所有質心並將結果存儲到列表中，該怎么辦。

Answer 1

Scikit-learn 不會為您提供中間結果，並且無法通過標准 API 來做到這一點。 獲得它們的一種黑客方法是使用這樣的東西：

k_means = KMeans(max_iter=1)
for i in range(300):
  k_means.fit(X)
  intermediate_centers = k_means.cluster_centers_
  k_means = KMeans(max_iter=1, init=intermediate_centers)

這不是一種快速的方法，我不建議在生產中運行它。