scikit-learn 1 · Training models
Cluster data
Group rows into k clusters without labels.
from sklearn.cluster import KMeans
kmeans = KMeans(n_clusters=3, n_init=10, random_state=42).fit(X)
clusters = kmeans.labels_
Official scikit-learn 1 documentation