可以使用 matplotlib 库绘制散点图来显示 kmeans 聚类的结果。

首先,需要将用户数据降维至二维空间,可以使用主成分分析(PCA)或 t-SNE 等降维方法。然后,根据聚类结果,将不同类别的用户用不同的颜色或标记绘制在散点图上。

以下是一个示例代码:

import matplotlib.pyplot as plt
from sklearn.decomposition import PCA

# 将用户数据降至二维空间
pca = PCA(n_components=2)
user_data_2d = pca.fit_transform(user_data)

# 绘制散点图
colors = ['r', 'g', 'b', 'y']
markers = ['o', 's', '^', 'd']
for i in range(len(user_km.cluster_centers_)):
    plt.scatter(user_data_2d[user_km.labels_ == i, 0], 
                user_data_2d[user_km.labels_ == i, 1], 
                c=colors[i], marker=markers[i], label='Cluster ' + str(i))
plt.scatter(user_centers.iloc[:, 0], user_centers.iloc[:, 1], 
            c='k', marker='*', s=150, label='Cluster Centers')
plt.legend()
plt.show()

其中,user_km.labels_ 表示每个用户被分配到的聚类标签,user_centers.iloc[:, 0]user_centers.iloc[:, 1] 分别表示聚类中心在降维后的二维空间中的坐标。可以根据具体情况修改颜色、标记和标签等参数。


原文地址: https://www.cveoy.top/t/topic/oRk0 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录