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

时间:2017-10-20 07:56:22

标签: python visualization k-means

我在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的簇。我如何可视化这些簇?

1 个答案:

答案 0 :(得分:6)

您可以使用pandas绘图工具parallel_coordinates显示多维聚类。

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