0001_clustering

0001_clustering

from pylab import plt, mpl
from sklearn.cluster import KMeans
from sklearn.datasets import make_blobs

x, y = make_blobs(n_samples=200, centers=5,
                          random_state=500, cluster_std=1.25)

model = KMeans(n_clusters=5, random_state=0)
model.fit(x)
KMeans(n_clusters=5, random_state=0)
y_ = model.predict(x)

plt.figure(figsize=(10, 6))
plt.scatter(x[:, 0], x[:, 1], c=y_,  cmap='coolwarm');