k-means

import os
import ipdb; # ipdb.set_trace()

import numpy as np

from sklearn.cluster import KMeans


X = np.array([[1, 2], [1, 4], [1, 0], [10, 2], [10, 4], [10, 0]])
kmeans = KMeans(n_clusters=2).fit(X)

# print the results for the training data
print(kmeans.labels_)

# predict for new data
print(kmeans.predict([[10, 4.5]]))
#print(kmeans.predict([[0,0], [12,3]]))

# show centers
print(kmeans.cluster_centers_)