如何在 sklearn k 均值聚类中将具有较大点数的组标记为“0”

Question

在sklearn中使用K-means聚类方法时，我将点聚类为两组。如何设置具有较大点数的组的k_means.labels_作为"0"（而不是 1）？

谢谢！

Answer 1

通常，如果您有完全标记的数据，则应使用分类器（请参阅此出色的图形）。K-means 是一个部分随机的过程，因此无法保证将哪个聚类分配给哪个标签。

获得预测后，如果要反转类标签，可以执行以下操作：

predictions = k_means.fit_predict( my_data )
if sum( predicitons==1 ) > sum( predictions==0 ):
    corrected_predictions = predictions.copy()
    corrected_predictions[ predictions==1 ] = 0
    corrected_predictions[ predictions==0 ] = 1

不建议使用

类的自动计算成员（如k_means.labels_）