Ошибка: графики данных алгоритма K-Mean Clustering не отображаются в Python - PullRequest
Новая
       14

Ошибка: графики данных алгоритма K-Mean Clustering не отображаются в Python

0 голосов
/ 12 октября 2018

Привет, я хочу реализовать алгоритм кластеризации K-Means.

Для этого я получаю данные из файла sample.csv и применяю к нему кластеризацию K-Means.Вот мой исходный код

## K-Means.py 

# clustering dataset
import pandas

from sklearn.cluster import KMeans
from sklearn import metrics
import numpy as np
import matplotlib.pyplot as plt

variables = pandas.read_csv("/Users/srikanth/Desktop/sample1.csv")
print(variables)
x1 = variables[['X']]
x2 = variables[['Y']]
print(x1)
print(x2)

plt.plot()
plt.xlim([0, 10])
plt.ylim([0, 10])
plt.title('Dataset')
plt.xlabel('X - Values')
plt.ylabel('Y - Values')
plt.scatter(x1, x2)
plt.show()

# create new plot and data
plt.plot()
X = np.array(list(zip(x1, x2))).reshape(len(x1), 2)
colors = ['b', 'g', 'r']
markers = ['o', 'v', 's']

# KMeans algorithm
K = 3
kmeans_model = KMeans(n_clusters=K).fit(X)

plt.plot()
for i, l in enumerate(kmeans_model.labels_):
    plt.plot(x1[i], x2[i], color=colors[l], marker=markers[l],ls='None')
    plt.xlim([0, 10])
    plt.ylim([0, 10])
    plt.show()

После того, как я запустил код выше в терминале, вывод будет выглядеть так:

Console Output

plots visualization

На приведенном выше рисунке не показаны кластерные данные, поэтому я хочу увидеть своиКластерные данные графически визуально.Как я могу это исправить.Я новичок в этой области.Спасибо

1 Ответ

0 голосов
/ 12 октября 2018 
from sklearn.cluster import KMeans
from sklearn import metrics
import numpy as np
import matplotlib.pyplot as plt

variables = pandas.read_csv("/Users/srikanth/Desktop/sample1.csv")
print(variables)
x1 = variables[['X']]
x2 = variables[['Y']]
plt.plot()
plt.xlim([150, 190])
plt.ylim([40, 90])
plt.title('Dataset')
plt.xlabel('X - Values')
plt.ylabel('Y - Values')
plt.scatter(x1, x2)
plt.show()

Разброс, который он производит для 10 точек:

enter image description here

Для кода, использующего модель кластеризации kmeans, вычерчение для каждой метки в модели, которая даст 10 сюжетов.Простое изменение лимитов должно творить чудеса.

...