rv 0.20.1

Random variables
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use std::f64;

use criterion::Criterion;
use criterion::{criterion_group, criterion_main};
use rand_distr::Beta;
use rv::traits::{HasDensity, Sampleable};

fn draw_rand_distr<R: rand::Rng>(rng: &mut R) -> f64 {
    let beta = Beta::new(5.0, 2.0).unwrap();
    rng.sample(beta)
}

fn draw_rv<R: rand::Rng>(mut rng: &mut R) -> f64 {
    let beta = rv::dist::Beta::new(5.0, 2.0).unwrap();
    beta.draw(&mut rng)
}

fn draw_2u<R: rand::Rng>(rng: &mut R) -> f64 {
    let a = rng.random::<f64>().powf(1.0 / 5.0);
    let b = rng.random::<f64>().powf(1.0 / 2.0);
    a / (a + b)
}

fn draw_2u_recip<R: rand::Rng>(rng: &mut R) -> f64 {
    let a = rng.random::<f64>().powf(5.0_f64.recip());
    let b = rng.random::<f64>().powf(2.0_f64.recip());

    a / (a + b)
}

fn bench_beta_draw(c: &mut Criterion) {
    let mut group = c.benchmark_group("beta_draw");
    group.bench_function("rand_distr", |b| {
        b.iter(|| {
            let mut rng = rand::rng();
            draw_rand_distr(&mut rng)
        });
    });
    group.bench_function("draw_rv", |b| {
        let mut rng = rand::rng();
        b.iter(|| draw_rv(&mut rng));
    });
    group.bench_function("draw_2_uniform", |b| {
        let mut rng = rand::rng();
        b.iter(|| draw_2u(&mut rng));
    });
    group.bench_function("draw_2_uniform_recip", |b| {
        let mut rng = rand::rng();
        b.iter(|| draw_2u_recip(&mut rng));
    });
}

fn bench_beta_vs_kumaraswamy_ln_pdf(c: &mut Criterion) {
    let mut group = c.benchmark_group("bench_beta_vs_kumaraswamy_pdf");
    group.bench_function("beta", |b| {
        let beta = rv::dist::Beta::new(1.0, 2.0).unwrap();
        b.iter(|| {
            let _x = beta.ln_f(&0.5_f64);
        });
    });
    group.bench_function("kumaraswamy", |b| {
        let kuma = rv::dist::Kumaraswamy::new(1.0, 2.0).unwrap();
        b.iter(|| {
            let _x = kuma.ln_f(&0.5_f64);
        });
    });
}

fn bench_beta_vs_kumaraswamy_draw(c: &mut Criterion) {
    let mut group = c.benchmark_group("bench_beta_vs_kumaraswamy_draw");
    group.bench_function("beta", |b| {
        let mut rng = rand::rng();
        let beta = rv::dist::Beta::new(1.0, 2.0).unwrap();
        b.iter(|| {
            let _x: f64 = beta.draw(&mut rng);
        });
    });
    group.bench_function("kumaraswamy", |b| {
        let mut rng = rand::rng();
        let kuma = rv::dist::Kumaraswamy::new(1.0, 2.0).unwrap();
        b.iter(|| {
            let _x: f64 = kuma.draw(&mut rng);
        });
    });
}

criterion_group!(
    beta_benches,
    bench_beta_draw,
    bench_beta_vs_kumaraswamy_ln_pdf,
    bench_beta_vs_kumaraswamy_draw,
);
criterion_main!(beta_benches);