klangner / hmm

Hidden Markov Model
Apache License 2.0
8 stars 1 forks source link
crates hidden-markov-model hmm viterbi-algorithm

Hidden Markov Model

Crates.io Crates.io docs.rs

hmm is an early-stage open-source project. It means that API can change at any time. If you think that this library can help you, then let me know. We can discuss future direction and try to stabilize the API.

Features

Example

Lets say that we have 2 coins:

We also know that after each toss we can switch coin with the probability of

First time we select coin with probability of 1/2

Using this library we can answer the following question:

Given the observations 'H H T T T' which coins were used during each toss?

Lest build HMM model and check the answer:

extern crate hmm;

use hmm::models::{HiddenMarkov};

fn main() {
    // 50%:50% chance to start at either coin
    let initials = vec![0.5, 0.5];

    // state transitions between coins:
    //   3/4 to stay on same coin (on matrix diagonals)
    //   1/4 chance to switch
    let st = vec![ vec![0.75, 0.25],
                   vec![0.25, 0.75]];

    // observation probabilities for each coin:
    //    first coin is fair, second one is biased
    let obs = vec![ vec![0.5, 0.5],
                    vec![0.25, 0.75]];

    // generate the HMM model
    let hmm = HiddenMarkov::new(initials, st, obs).unwrap();
    let coins = hmm.map_estimate(vec![0, 0, 1, 1, 1]);
    println!("Coins used: {:?}", coins);
}

For more check Examples.

License

Licensed under either of

at your option.

Contributions

Unless you explicitly state otherwise, any contribution intentionally submitted for inclusion in the work by you, as defined in the Apache-2.0 license, shall be dual licensed as above, without any additional terms or conditions.