Skip to content

Instantly share code, notes, and snippets.

@grohith327

grohith327/k_means_4.py

Created June 9, 2018 15:14
Show Gist options
  • Save grohith327/7fbb9aec40a336672836e30d2b081ce8 to your computer and use it in GitHub Desktop.
Save grohith327/7fbb9aec40a336672836e30d2b081ce8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
k_means_4.py
from sklearn.cluster import KMeans
clf = KMeans(n_clusters = 3)
clf.fit(train_data)
pred = clf.predict(test_data)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment