python - 来自 Scikit-Learn 的 K 均值的失真函数

Question

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

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

Answer 1

这已经在 KMeans 类的 inertia_ 属性中在 fit 时间预先计算。

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