class

Stochasta::GMM::Model

Inherits Reference < Object

Constructors

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

Instance methods

components

components=(components : Array(Component))

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

Fits the GMM on the dataset using Expectation-Maximization

k

k=(k : Int32)

max_iter

max_iter=(max_iter : Int32)

predict(x : Array(Float64)) : Int32

Predicts the most likely cluster (hard assignment) for a given point

predict_probabilities(x : Array(Float64)) : Array(Float64)

Predicts the responsibilities (soft assignments) for a given point

tolerance

tolerance=(tolerance : Float64)