Struct GaussianHMM 

pub struct GaussianHMM {
    pub n_components: usize,
    pub n_features: usize,
    pub covariance_type: HMMCovarianceType,
    pub max_iter: usize,
    pub tol: f32,
    pub n_init: usize,
    /* private fields */
}

Hidden Markov Model with Gaussian emissions

Fields

n_components: usize

n_features: usize

covariance_type: HMMCovarianceType

max_iter: usize

tol: f32

n_init: usize

Implementations

impl GaussianHMM

pub fn new(n_components: usize, n_features: usize) -> Self

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

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

pub fn fit(&mut self, sequences: &[Tensor])

Fit the HMM using Baum-Welch algorithm (EM for HMMs)

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

Predict hidden state sequence using Viterbi algorithm