More than 5 years have passed since last update.

scikit-learnでkmeans++

Python
scikit-learn

Posted at 2016-11-16

動機

自前のクラスタリング手法においてkmeans++を使いたい．
sklearn.cluster.KMeansの初期化方法にkmeans++があるので，これをなんとか呼び出したい．

できる


from sklearn.cluster.k_means_ import _k_init
from sklearn.utils.extmath import row_norms
import numpy as np

# x: データ, k: クラスタ数
c = _k_init(x, k, row_norms(x, squared=True), np.random.RandomState())

You get articles that match your needs
You can efficiently read back useful information
You can use dark theme

What you can do with signing up