Struct GaussianMixture 

pub struct GaussianMixture {
    pub n_components: usize,
    pub covariance_type: CovarianceType,
    pub max_iter: usize,
    pub tol: f32,
    pub reg_covar: f32,
    pub n_init: usize,
    /* private fields */
}

Gaussian Mixture Model

Fields

n_components: usize

covariance_type: CovarianceType

max_iter: usize

tol: f32

reg_covar: f32

n_init: usize

Implementations

impl GaussianMixture

pub fn new(n_components: usize) -> Self

pub fn covariance_type(self, cov_type: CovarianceType) -> Self

pub fn max_iter(self, iter: usize) -> Self

pub fn tol(self, tolerance: f32) -> Self

pub fn fit(&mut self, x: &Tensor)

Fit the Gaussian Mixture Model using EM algorithm

pub fn predict(&self, x: &Tensor) -> Tensor

Predict cluster labels

pub fn predict_proba(&self, x: &Tensor) -> Tensor

Predict probabilities for each component

pub fn sample(&self, n_samples: usize) -> Tensor

Sample from the fitted model