rust_metrics 0.1.4

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

#![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,
    BinaryPrecision, BinaryRecall, MulticlassAccuracy, MulticlassF1Score, MulticlassHingeLoss,
    MulticlassPrecision,
};
pub use core::{Metric, MetricError};
pub use regression::{MeanAbsoluteError, MeanAbsolutePercentageError, MeanSquaredError, 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;