modppl 0.3.1

a experimental library for probabilistic programming in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use super::{u01, Distribution};
use crate::Real;
use rand::rngs::ThreadRng;

/// Bernoulli distribution type
pub struct Bernoulli {}

/// Instantiation of the Bernoulli distribution
pub const bernoulli: Bernoulli = Bernoulli {};

impl Distribution<bool, Real> for Bernoulli {
    fn logpdf(&self, a: &bool, p: Real) -> Real {
        (if *a { p } else { 1. - p }).ln()
    }

    fn random(&self, rng: &mut ThreadRng, p: Real) -> bool {
        p > u01(rng)
    }
}