2017-10-20 712 views
0

我在mnist数据集上使用kmeans聚类算法,并希望在聚类后可视化绘图。到目前为止,我这样做如何可视化多维数据上的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_ 

所以,我现在有代表0到9。我怎么能想象这些集群10个集群?

回答

0

您可以使用熊猫绘图工具parallel_coordinates可视化多维聚类。

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