KMeans Clustering

89 阅读2分钟
import numpy as np
from sklearn.cluster import KMeans
from sklearn.datasets import make_blobs
n_samples = 300
random_state = 20
X,y = make_blobs(n_samples=n_samples,n_features=5,random_state=None)
predict_y = KMeans(n_clusters=3,random_state=random_state).fit_predict(X)
predict_y
array([2, 0, 0, 1, 1, 2, 1, 2, 1, 1, 0, 0, 1, 2, 2, 2, 2, 1, 0, 2, 0, 2,
       2, 1, 1, 2, 2, 1, 2, 1, 1, 2, 2, 2, 0, 2, 1, 0, 2, 1, 0, 2, 0, 2,
       2, 1, 2, 2, 0, 1, 1, 2, 2, 1, 0, 2, 1, 0, 1, 1, 1, 0, 2, 1, 0, 1,
       0, 1, 0, 1, 2, 1, 2, 1, 2, 0, 1, 0, 1, 0, 2, 1, 0, 2, 1, 0, 1, 0,
       1, 2, 0, 1, 0, 2, 1, 0, 2, 1, 2, 0, 2, 0, 1, 1, 0, 2, 0, 2, 2, 1,
       0, 0, 1, 2, 1, 0, 0, 2, 2, 1, 0, 1, 1, 2, 0, 1, 1, 2, 1, 0, 2, 2,
       2, 1, 1, 0, 2, 1, 0, 2, 0, 0, 2, 2, 2, 0, 0, 2, 1, 2, 2, 1, 0, 2,
       2, 1, 0, 0, 2, 1, 0, 0, 0, 2, 1, 1, 1, 0, 1, 1, 1, 0, 0, 0, 2, 2,
       0, 0, 0, 1, 1, 1, 1, 2, 1, 2, 1, 0, 0, 2, 0, 0, 2, 0, 2, 1, 2, 0,
       1, 0, 2, 1, 1, 2, 1, 1, 0, 2, 0, 0, 0, 0, 1, 2, 1, 2, 1, 0, 1, 0,
       0, 1, 1, 0, 0, 1, 0, 1, 1, 1, 2, 2, 2, 2, 0, 0, 1, 1, 0, 0, 1, 2,
       0, 2, 2, 1, 0, 1, 0, 0, 0, 0, 0, 1, 0, 2, 2, 2, 1, 2, 2, 2, 0, 1,
       2, 1, 2, 0, 2, 2, 2, 0, 1, 2, 0, 1, 2, 0, 0, 1, 0, 1, 2, 2, 2, 0,
       0, 1, 1, 0, 1, 0, 1, 0, 2, 2, 0, 2, 0, 2])
y
array([0, 1, 1, 2, 2, 0, 2, 0, 2, 2, 1, 1, 2, 0, 0, 0, 0, 2, 1, 0, 1, 0,
       0, 2, 2, 0, 0, 2, 0, 2, 2, 0, 0, 0, 1, 0, 2, 1, 0, 2, 1, 0, 1, 0,
       0, 2, 0, 0, 1, 2, 2, 0, 0, 2, 1, 0, 2, 1, 2, 2, 2, 1, 0, 2, 1, 2,
       1, 2, 1, 2, 0, 2, 0, 2, 0, 1, 2, 1, 2, 1, 0, 2, 1, 0, 2, 1, 2, 1,
       2, 0, 1, 2, 1, 0, 2, 1, 0, 2, 0, 1, 0, 1, 2, 2, 1, 0, 1, 0, 0, 2,
       1, 1, 2, 0, 2, 1, 1, 0, 0, 2, 1, 2, 2, 0, 1, 2, 2, 0, 2, 1, 0, 0,
       0, 2, 2, 1, 0, 2, 1, 0, 1, 1, 0, 0, 0, 1, 1, 0, 2, 0, 0, 2, 1, 0,
       0, 2, 1, 1, 0, 2, 1, 1, 1, 0, 2, 2, 2, 1, 2, 2, 2, 1, 1, 1, 0, 0,
       1, 1, 1, 2, 2, 2, 2, 0, 2, 0, 2, 1, 1, 0, 1, 1, 0, 1, 0, 2, 0, 1,
       2, 1, 0, 2, 2, 0, 2, 2, 1, 0, 1, 1, 1, 1, 2, 0, 2, 0, 2, 1, 2, 1,
       1, 2, 2, 1, 1, 2, 1, 2, 2, 2, 0, 0, 0, 0, 1, 1, 2, 2, 1, 1, 2, 0,
       1, 0, 0, 2, 1, 2, 1, 1, 1, 1, 1, 2, 1, 0, 0, 0, 2, 0, 0, 0, 1, 2,
       0, 2, 0, 1, 0, 0, 0, 1, 2, 0, 1, 2, 0, 1, 1, 2, 1, 2, 0, 0, 0, 1,
       1, 2, 2, 1, 2, 1, 2, 1, 0, 0, 1, 0, 1, 0])