Struct GaussianMixture 

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

Gaussian Mixture Model using EM algorithm

Fields

n_components: usize

covariance_type: CovarianceType

max_iter: usize

tol: f32

n_init: usize

reg_covar: f32

Implementations

impl GaussianMixture

pub fn new(n_components: usize) -> Self

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

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

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

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

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

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

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

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

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