sklearn.cluster.kmeans_plusplus(X, n_clusters, *, x_squared_norms=None, random_state=None, n_local_trials=None)[source]

Init n_clusters seeds according to k-means++

New in version 0.24.

X{array-like, sparse matrix} of shape (n_samples, n_features)

The data to pick seeds from.


The number of centroids to initialize

x_squared_normsarray-like of shape (n_samples,), default=None

Squared Euclidean norm of each data point.

random_stateint or RandomState instance, default=None

Determines random number generation for centroid initialization. Pass an int for reproducible output across multiple function calls. See Glossary.

n_local_trialsint, default=None

The number of seeding trials for each center (except the first), of which the one reducing inertia the most is greedily chosen. Set to None to make the number of trials depend logarithmically on the number of seeds (2+log(k)).

centersndarray of shape (n_clusters, n_features)

The inital centers for k-means.

indicesndarray of shape (n_clusters,)

The index location of the chosen centers in the data array X. For a given index and center, X[index] = center.


Selects initial cluster centers for k-mean clustering in a smart way to speed up convergence. see: Arthur, D. and Vassilvitskii, S. “k-means++: the advantages of careful seeding”. ACM-SIAM symposium on Discrete algorithms. 2007


>>> from sklearn.cluster import kmeans_plusplus
>>> import numpy as np
>>> X = np.array([[1, 2], [1, 4], [1, 0],
...               [10, 2], [10, 4], [10, 0]])
>>> centers, indices = kmeans_plusplus(X, n_clusters=2, random_state=0)
>>> centers
array([[10,  4],
       [ 1,  0]])
>>> indices
array([4, 2])

Examples using sklearn.cluster.kmeans_plusplus