frrostbitee

KMeans

Apr 15th, 2019
49
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.80 KB | None | 0 0
  1.     dataset = pd.read_csv('test.csv')
  2.     X = dataset.iloc[:,:].values
  3.    
  4.     wcss = []
  5.     for i in range(1,11):
  6.         kmeans = KMeans(n_clusters = i, init= 'k-means++', max_iter=300, n_init=10, random_state=0)
  7.         kmeans.fit(X)
  8.         wcss.append(kmeans.inertia_)
  9.     plt.plot(range(1,11),wcss)
  10.     #plt.show()
  11.     kmeans = KMeans(n_clusters =3, init= 'k-means++', max_iter=300, n_init=10, random_state=0)
  12.     y_kmeans = kmeans.fit_predict(X)
  13.     plt.scatter(X[y_kmeans==0,0],X[y_kmeans==0,1], s=100, c='red', label='One')
  14.     plt.scatter(X[y_kmeans==1,0],X[y_kmeans==1,1], s=100, c='green', label='Two')
  15.     plt.scatter(X[y_kmeans==2,0],X[y_kmeans==2,1], s=100, c='cyan', label='Three')
  16.     plt.scatter(kmeans.cluster_centers_[:,0],kmeans.cluster_centers_[:,1],s=300, c='yellow',label='centroids')
Add Comment
Please, Sign In to add comment