Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- KMEANS_CLUSTERS = 2
- ...
- model = KMeans(
- n_clusters=KMEANS_CLUSTERS,
- init='k-means++',
- max_iter=5,
- n_init=1,
- verbose=True
- )
- groups = model.fit_predict(data)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement