rhmm 0.0.2

A rust implementation of hidden markov models.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

//! Common types used throughout the library

use ndarray::{Array1, Array2};

/// Type alias for state transition matrix
pub type TransitionMatrix = Array2<f64>;

/// Type alias for initial state probabilities
pub type InitialProbs = Array1<f64>;

/// Type alias for observation sequences
pub type Observations = Array2<f64>;

/// Type alias for state sequences
pub type States = Array1<usize>;

/// Covariance type for Gaussian models
#[derive(Debug, Clone, Copy, PartialEq, Default)]
pub enum CovarianceType {
    /// Full covariance matrix
    Full,
    /// Diagonal covariance matrix
    #[default]
    Diagonal,
    /// Spherical covariance (single variance value)
    Spherical,
    /// Tied covariance (same for all states)
    Tied,
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_covariance_type_default() {
        assert_eq!(CovarianceType::default(), CovarianceType::Diagonal);
    }

    #[test]
    fn test_covariance_type_equality() {
        assert_eq!(CovarianceType::Full, CovarianceType::Full);
        assert_ne!(CovarianceType::Full, CovarianceType::Diagonal);
    }

    #[test]
    fn test_type_aliases() {
        let transition: TransitionMatrix = Array2::zeros((3, 3));
        assert_eq!(transition.shape(), &[3, 3]);

        let initial: InitialProbs = Array1::zeros(3);
        assert_eq!(initial.len(), 3);

        let obs: Observations = Array2::zeros((10, 2));
        assert_eq!(obs.shape(), &[10, 2]);

        let states: States = Array1::zeros(10);
        assert_eq!(states.len(), 10);
    }
}