如何在多维数据上可视化 kmeans 聚类



我在 mnist 数据集上使用 kmeans 聚类算法,并希望在聚类后可视化图。到目前为止,我做到了

from mnist import MNIST
mndata = MNIST('Datasets')
X_train, y_train = mndata.load_training()
#do the clustering
k_means = cluster.KMeans(n_clusters=len(np.unique(y_train)))
k_means.fit(X_train)
labels = k_means.labels_

因此,我现在有 10 个聚类,代表 0 到 9。如何可视化这些群集?

您可以使用

熊猫绘图工具parallel_coordinates可视化多维聚类。

predict = k_means.predict(data)
data['cluster'] = predict
pandas.tools.plotting.parallel_coordinates(data, 'cluster')

最新更新