plotters-statistical 0.1.0

Statistical chart primitives (box, violin, ROC, PR, regularization-path, residual) as native plotters series
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

//! Generic annotated heatmap (here, a confusion matrix) rendered to
//! `heatmap.svg`, using a custom colormap override.

use plotters::prelude::*;
use plotters_statistical::colormap::{GradientColorMap, Normalization};
use plotters_statistical::series::heatmap::HeatmapAnnotation;
use plotters_statistical::Heatmap;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let root = SVGBackend::new("heatmap.svg", (560, 520)).into_drawing_area();
    root.fill(&WHITE)?;

    // A 3x3 confusion matrix (row = true class, col = predicted).
    let counts = vec![
        vec![52.0, 3.0, 1.0],
        vec![4.0, 43.0, 6.0],
        vec![0.0, 5.0, 61.0],
    ];

    let mut chart = ChartBuilder::on(&root)
        .caption("Confusion matrix", ("sans-serif", 22))
        .margin(20)
        .set_label_area_size(LabelAreaPosition::Left, 40)
        .set_label_area_size(LabelAreaPosition::Bottom, 40)
        .build_cartesian_2d(-0.5f64..2.5f64, -0.5f64..2.5f64)?;
    chart
        .configure_mesh()
        .disable_mesh()
        .x_desc("predicted")
        .y_desc("true")
        .draw()?;

    chart.draw_series(std::iter::once(
        Heatmap::new(&counts)
            .colormap(GradientColorMap::blues())
            .normalization(Normalization::Linear {
                min: 0.0,
                max: 61.0,
            })
            .annotate(HeatmapAnnotation {
                precision: 0,
                font_size: 16,
                text_color: None,
            })
            .cell_gap(1),
    ))?;

    root.present()?;
    println!("wrote heatmap.svg");
    Ok(())
}