def test(self, k, data, random=False):
kmeans = KMeans()
book = kmeans.iterate(k, data, random)
print('---Start---')
for i in range(len(book)):
b = book[i]
print(b[0])
listX = []
listY = []
repVecX = [b[0][0]]
repVecY = [b[0][1]]
for vec in b[1]:
listX.append(vec[0])
listY.append(vec[1])
plt.plot(listDX, listDY, 'ro', listX, listY, 'g^', repVecX,
repVecY, 'bs')
plt.axis([-5, 30, -5, 30])
plt.show()
plt.clf()
print('---End---')
from create_cluster_data import create_cluster_data import maths as mth from KMeans import KMeans from sklearn.preprocessing import scale data = create_cluster_data(1000, 5) kmean = KMeans(mth.standardize(data)) kmean.fit() kmean.iterate() kmean.plot()
