我正在尝试在 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:宝马、欧宝
我怎样才能做到这一点 ?