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