Guest User

Untitled

a guest
May 26th, 2018
85
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.30 KB | None | 0 0
  1. def doKmeans(X, nclust):
  2. model = KMeans(nclust)
  3. model.fit(X)
  4. clust_labels = model.predict(X)
  5. cent = model.cluster_centers_
  6. return (clust_labels, cent)
  7.  
  8. clust_labels, cent = doKmeans(df_new, 4)
  9. kmeans = pd.DataFrame(clust_labels)
  10.  
  11. df_new.insert((df_new.shape[1]),'clusterNo',kmeans)
Add Comment
Please, Sign In to add comment