Advertisement
Guest User

Untitled

a guest
Mar 26th, 2019
67
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.72 KB | None | 0 0
  1. import numpy as np
  2. from sklearn.cluster import KMeans
  3. from sklearn import metrics
  4. import matplotlib.pyplot as plt
  5. plt.subplot(3, 2, 1)
  6.  
  7. x1 = [15, 19, 15, 5, 13, 17, 15, 12, 8, 6, 9, 13]
  8. x2 = [13, 16, 17, 6, 17, 14, 15, 13, 7, 6, 4, 12]
  9. plt.title('Instances')
  10. plt.scatter(x1, x2)
  11.  
  12. X = np.array(list(zip(x1, x2)))
  13.  
  14. c = ['b', 'g', 'r', 'c', 'm', 'y', 'k', 'b']
  15. m = ['o', 's', 'D', 'v', '^', 'p', '*', '+']
  16.  
  17. p = 1
  18. for t in [2, 3, 4, 5, 8]:
  19. p += 1
  20. plt.subplot(3, 2, p)
  21. kmeans_model = KMeans(n_clusters=t).fit(X)
  22. for i, j in enumerate(kmeans_model.labels_):
  23. plt.plot(x1[i], x2[i], color=c[j], marker=m[j],ls='None')
  24. plt.title('K = %s, SC = %.03f' % (t, metrics.silhouette_score(X, kmeans_model.labels_,metric='euclidean')))
  25. plt.show()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement