rust_metrics 0.1.5

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
20
21

//! 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 jaccard;
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, MulticlassF1Score};
pub use hinge::{BinaryHingeLoss, MulticlassHingeLoss};
pub use jaccard::{BinaryJaccardIndex, MulticlassJaccardIndex};
pub use precision_recall::{BinaryPrecision, BinaryRecall, MulticlassPrecision};