Struct rusty_machine::learning::gmm::GaussianMixtureModel

pub struct GaussianMixtureModel {
    pub cov_option: CovOption,
    // some fields omitted
}

A Gaussian Mixture Model

Fields

cov_option: CovOption

The covariance options for the GMM.

Methods

impl GaussianMixtureModel

fn new(k: usize) -> GaussianMixtureModel

Constructs a new Gaussian Mixture Model

Defaults to 100 maximum iterations and full covariance structure.

Examples

use rusty_machine::learning::gmm::GaussianMixtureModel;

let gmm = GaussianMixtureModel::new(3);

fn set_max_iters(&mut self, iters: usize)

Sets the max number of iterations for the EM algorithm.

Examples

use rusty_machine::learning::gmm::GaussianMixtureModel;

let mut gmm = GaussianMixtureModel::new(2);
gmm.set_max_iters(5);

Trait Implementations

impl UnSupModel<Matrix<f64>, Matrix<f64>> for GaussianMixtureModel

fn train(&mut self, inputs: &Matrix<f64>)

Train the model using inputs.

fn predict(&self, inputs: &Matrix<f64>) -> Matrix<f64>

Predict output from inputs.