代码之家 › 专栏 › 技术社区 › Evgeni Nabokov

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

mean-shift scikit-learn python

-1

Evgeni Nabokov · 技术社区 · 6 年前

我有一个受过训练的MeanShift对象( ms ). 它有一个简单的中心列表。如何确定中心所属的标签? 我知道 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 ]]

1 回复 | 直到 6 年前

1

Matthieu Brucher 6 年前

维度 labels i 在原始数据中: cluster_centers_[labels_[i]] .

labels == k 选择带有该标签的所有数据( X[labels_ == k] https://scikit-learn.org/stable/auto_examples/cluster/plot_mean_shift.html#sphx-glr-auto-examples-cluster-plot-mean-shift-py