rust_metrics 0.1.3

Incremental evaluation metrics for classification and text generation pipelines.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! Classification metrics inspired by TorchMetrics.
//!
//! Every struct in this module implements [`Metric`](crate::core::Metric) and therefore supports
//! batched updates plus `reset`/`compute` semantics.

pub mod accuracy;
pub mod auroc;
pub mod confusion_matrix;
pub mod f1;
pub mod hinge;
pub mod precision_recall;
pub mod stat_scores;

pub use accuracy::{BinaryAccuracy, MulticlassAccuracy};
pub use auroc::BinaryAuroc;
pub use confusion_matrix::BinaryConfusionMatrix;
pub use f1::BinaryF1Score;
pub use hinge::BinaryHinge;
pub use precision_recall::{BinaryPrecision, BinaryRecall};