KMeans
KMeans
A model type for constructing a k means, based on MLJScikitLearnInterface.jl, and implementing the MLJ model interface.
From MLJ, the type can be imported using
KMeans = @load KMeans pkg=MLJScikitLearnInterface
Do model = KMeans()
to construct an instance with default hyper-parameters. Provide keyword arguments to override hyper-parameter defaults, as in KMeans(n_clusters=...)
.
K-Means algorithm: find K centroids corresponding to K clusters in the data.