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
22
23
24
25

#![doc = include_str!("../README.md")]

pub mod classification;
pub mod core;
pub mod regression;
pub mod text;
pub mod utils;

pub use classification::{
    BinaryAccuracy, BinaryAuroc, BinaryConfusionMatrix, BinaryF1Score, BinaryHingeLoss,
    BinaryJaccardIndex, BinaryPrecision, BinaryRecall, MulticlassAccuracy, MulticlassF1Score,
    MulticlassHingeLoss, MulticlassJaccardIndex, MulticlassPrecision,
};
pub use core::{Metric, MetricError};
pub use regression::{
    MeanAbsoluteError, MeanAbsolutePercentageError, MeanSquaredError,
    NormalizedRootMeanSquaredError, R2Score,
};

pub use text::{Bleu, EditDistance};
pub use utils::Reduction;

#[cfg_attr(docsrs, doc(cfg(feature = "text-bert")))]
#[cfg(feature = "text-bert")]
pub use text::SentenceEmbeddingSimilarity;