【Python-ML】聚类的性能评价指标

最新推荐文章于 2024-06-09 12:14:43 发布

fjssharpsword

最新推荐文章于 2024-06-09 12:14:43 发布

阅读量5.4k

点赞数

分类专栏： Big data

本文链接：https://blog.csdn.net/fjssharpsword/article/details/79197226

版权

参考：http://scikit-learn.org/stable/modules/clustering.html#clustering-performance-evaluation

1、兰德指数

from sklearn import metrics
labels_true = [0, 0, 0, 1, 1, 1]
labels_pred = [0, 0, 1, 1, 2, 2]

print (metrics.adjusted_rand_score(labels_true, labels_pred))

2、互信息

from sklearn import metrics
labels_true = [0, 0, 0, 1, 1, 1]
labels_pred = [0, 0, 1, 1, 2, 2]
print (metrics.adjusted_mutual_info_score(labels_true, labels_pred) )