我正在尝试在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]]
每个数据都有一个标签。例如:
[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)
我的目标是在运行KMeans后,获取每个聚类的标签。
以下是一个虚假的例子:
聚类1: Fiat1, Fiat2
聚类2: Mercedes
聚类3: bmw, Opel
我该怎么办?