Как визуализировать кластеризацию kmeans на многомерных данных

Question

Я использую алгоритм кластеризации kmeans для набора данных mnist и хочу визуализировать графики после кластеризации. Пока я сделал это

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. Как я могу визуализировать эти кластеры?

Answer 1

Вы можете визуализировать многомерную кластеризацию, используя инструмент для построения графиков pandas parallel_coordinates.

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