lace 0.9.1

A probabilistic cross-categorization engine
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

#[cfg(feature = "formats")]
fn main() {
    use std::io::Write;

    use lace::prelude::*;
    use lace::rv::prelude::*;

    let mut rng = rand::rng();

    // Draw data from a mixture of Poisson
    let mixture = Mixture::uniform(vec![
        Poisson::new(3.0).unwrap(),
        Poisson::new(10.0).unwrap(),
    ])
    .unwrap();

    let mut engine = {
        let mut file = tempfile::NamedTempFile::new().unwrap();
        writeln!(file, "id,x").unwrap();
        mixture
            .sample_stream(&mut rng)
            .take(500)
            .enumerate()
            .for_each(|(ix, x): (usize, u32)| {
                writeln!(file, "{},{}", ix, x).unwrap();
            });

        EngineBuilder::new(DataSource::Csv(file.path().into()))
            .with_nstates(2)
            .seed_from_u64(1337)
            .build()
            .unwrap()
    };

    engine.run(500).unwrap();

    let vals: Vec<_> = (0_u32..30).map(|x| vec![Datum::Count(x)]).collect();

    let fx: Vec<_> = engine
        .logp(&[0], &vals, &Given::<usize>::Nothing, None)
        .unwrap()
        .iter()
        .map(|ln_f| ln_f.exp())
        .collect();

    println!("x,fx_true,fx_lace");
    for (x, fx) in fx.iter().enumerate() {
        println!("{},{},{}", x, mixture.f(&(x as u32)), fx);
    }
}

#[cfg(not(feature = "formats"))]
fn main() {
    eprintln!("Please enable the 'formats' feature to use this example.")
}