小编dij*_*iri的帖子

向 sklearn k-means 添加标签

我正在尝试在 python 中使用 kmeans。

data = [[1,2,3,4,5],[1,0,3,2,4],[4,3,234,5,5],[23,4,5,1,4],[23,5,2,3,5]]
Run Code Online (Sandbox Code Playgroud)

每个数据都有一个标签。例子:

[1,2,3,4,5] -> Fiat1
[1,0,3,2,4] -> Fiat2
[4,3,234,5,5] -> Mercedes
[23,4,5,1,4] -> Opel
[23,5,2,3,5] -> bmw

kmeans = KMeans(init='k-means++', n_clusters=3, n_init=10)
kmeans.fit(data)
Run Code Online (Sandbox Code Playgroud)

我的目标是在运行 KMeans 之后,我想获得每个集群的标签。

一个假的例子:

集群 1:Fiat1、Fiat2

集群 2:梅赛德斯

集群 3:宝马、欧宝

我怎样才能做到这一点 ?

python scikit-learn

5
推荐指数
1
解决办法
6473
查看次数

标签 统计

python ×1

scikit-learn ×1