esoc-chart 0.1.0

High-level charting API built on esoc-gfx — matplotlib-equivalent for Rust
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
56
57

// SPDX-License-Identifier: MIT OR Apache-2.0
//! ROC curve visualization.

use crate::grammar::chart::Chart;
use crate::grammar::layer::{Layer, MarkType};
use crate::new_theme::NewTheme;

/// Extension trait for creating ROC curve figures.
pub trait RocCurveExt {
    /// Create a chart showing this ROC curve.
    fn roc_figure(&self) -> Chart;

    /// Create a chart showing this ROC curve with a custom theme.
    fn roc_figure_with_theme(&self, theme: NewTheme) -> Chart;
}

impl RocCurveExt for scry_learn::metrics::RocCurve {
    fn roc_figure(&self) -> Chart {
        self.roc_figure_with_theme(NewTheme::default())
    }

    fn roc_figure_with_theme(&self, theme: NewTheme) -> Chart {
        let auc_label = if self.auc.is_nan() {
            "ROC (AUC = N/A)".to_string()
        } else {
            format!("ROC (AUC = {:.3})", self.auc)
        };

        // ROC curve layer
        let roc_layer = Layer::new(MarkType::Line)
            .with_x(self.fpr.clone())
            .with_y(self.tpr.clone())
            .with_label("ROC Curve");

        // Diagonal reference line
        let diag_layer = Layer::new(MarkType::Line)
            .with_x(vec![0.0, 1.0])
            .with_y(vec![0.0, 1.0])
            .with_label("Random Classifier");

        Chart::new()
            .layer(roc_layer)
            .layer(diag_layer)
            .title(auc_label)
            .x_label("False Positive Rate")
            .y_label("True Positive Rate")
            .x_domain(0.0, 1.0)
            .y_domain(0.0, 1.0)
            .theme(theme)
            .size(600.0, 600.0)
    }
}

/// Create a ROC curve chart from an `RocCurve`.
pub fn roc_curve_figure(roc: &scry_learn::metrics::RocCurve) -> Chart {
    roc.roc_figure()
}