python - 如何在 sklearn.cluster.MeanShift 中获取特定集群的中心

Question

我有一个训练有素的 MeanShift 对象 ( ms)。它有一个简单的中心列表。如何确定一个中心所属的标签？我知道labels_，但我没有看到 和 之间的labels_联系cluster_centers_。

print(ms.cluster_centers_)

[[  40.7177164   -73.99183542]
 [  33.44943805 -112.00213969]
 [  33.44638027 -111.90188756]
 ..., 
 [  46.7323875  -117.0001651 ]
 [  29.6899563   -95.8996757 ]
 [  31.3787916   -95.3213317 ]]

Answer 1

的维度labels是原始数据集的维度。它给出了相应簇的索引。i所以原始数据中一个条目的关联聚类中心是cluster_centers_[labels_[i]]。

您可以在 sklearn 的示例中看到它们循环使用唯一标签的数量，并使用该标签 ( ) labels == k选择所有数据： https://scikit-learn.org/stable/auto_examples/cluster/plot_mean_shift。 html#sphx-glr-auto-examples-cluster-plot-mean-shift-pyX[labels_ == k]