来自K Means of Scikit-Learn的失真函数

时间:2012-07-17 19:30:40

标签: python k-means scikit-learn

要获得失真函数(每个点到其中心的距离之和),当K表示通过Scikit-Learn进行聚类时,一种简单的方法就是获取中心(k_means.cluster_centers_)并总结距离每一点。

只是想知道是否有更快的方法? (就程序员时间而言)类似于直接函数调用等。

1 个答案:

答案 0 :(得分:2)

fit类的inertia_属性已在KMeans时预先计算出来。

>>> from sklearn.datasets import load_iris
>>> from sklearn.cluster import KMeans
>>> iris = load_iris()
>>> km = KMeans(3).fit(iris.data)
>>> km.inertia_
78.940841426146108