Contoh KMeans Python

  Umum
from sklearn.cluster import KMeans
import numpy as np
X = np.array([[1, 2], [1, 4], [1, 0],
              [10, 2], [10, 4], [10, 0]])
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
label = kmeans.labels_
print(label)


prediksi = kmeans.predict([[0, 0], [12, 3]])
print(prediksi)


kmeans.cluster_centers_

LEAVE A COMMENT