MiniBatchKMeans
MiniBatchKMeans
A model type for constructing a Mini-Batch K-Means clustering., based on MLJScikitLearnInterface.jl, and implementing the MLJ model interface.
From MLJ, the type can be imported using
MiniBatchKMeans = @load MiniBatchKMeans pkg=MLJScikitLearnInterface
Do model = MiniBatchKMeans()
to construct an instance with default hyper-parameters. Provide keyword arguments to override hyper-parameter defaults, as in MiniBatchKMeans(n_clusters=...)
.
Hyper-parameters
n_clusters = 8
max_iter = 100
batch_size = 100
verbose = 0
compute_labels = true
random_state = nothing
tol = 0.0
max_no_improvement = 10
init_size = nothing
n_init = 3
init = k-means++
reassignment_ratio = 0.01