class

Stochasta::KMeans::Model

Inherits Reference < Object

Constructors

new(k : Int32, max_iter : Int32 = 300, tolerance : Float64 = 1e-4)

Instance methods

assignments

assignments=(assignments : Array(Int32))

centroids

centroids=(centroids : Array(Array(Float64)))

fit(data : Array(Array(Float64))) : self

Fits the data and clusters them

k

k=(k : Int32)

max_iter

max_iter=(max_iter : Int32)

predict(point : Array(Float64)) : Int32

Predict the closest cluster for a new data point

tolerance

tolerance=(tolerance : Float64)