在K均值聚类之后绘制聚类

时间:2020-05-12 08:21:43

标签: python scikit-learn k-means

我在数据集上使用了K-Means聚类,发现了7个聚类,但无法绘制聚类。请在下面找到我使用的代码:-

# Plot the clustered data
fig, ax = plt.subplots(figsize=(10, 10))
plt.scatter(general_population_scaled[kmeans.labels == 0, 0], general_population_scaled[kmeans.labels == 0, 1],
            c='green', label='cluster 1')

请帮助我更正代码以生成结果。这是我得到的错误:-

AttributeError: 'KMeans' object has no attribute 'labels'

1 个答案:

答案 0 :(得分:0)

集群标签位于labels_对象的kmeans属性中,而不是labels属性

https://scikit-learn.org/stable/modules/generated/sklearn.cluster.KMeans.html