Advertisement
Guest User

Untitled

a guest
Jun 24th, 2019
89
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.33 KB | None | 0 0
  1. from sklearn.cluster import KMeans, MiniBatchKMeans
  2. km = KMeans(n_clusters=5)
  3. k_range = range(1,10)
  4. sse = []
  5. for k in k_range:
  6. km = KMeans(n_clusters=k).fit(dataset_to_predict) #this line throw error
  7. sse.append(km.inertia_)
  8.  
  9. y_predicted = km.fit_predict(dataset_to_predict) #this line throw error
  10. y_predicted
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement