Struct TextClassificationMetrics 

pub struct TextClassificationMetrics;

Text classification metrics

Implementations

impl TextClassificationMetrics

pub fn new() -> Self

Create a new metrics calculator

pub fn precision<T>( &self, predictions: &[T], true_labels: &[T], class_idx: Option<T>, ) -> Result<f64>

where T: PartialEq + Copy + Default,

Calculate precision score

pub fn recall<T>( &self, predictions: &[T], true_labels: &[T], class_idx: Option<T>, ) -> Result<f64>

where T: PartialEq + Copy + Default,

Calculate recall score

pub fn f1_score<T>( &self, predictions: &[T], true_labels: &[T], class_idx: Option<T>, ) -> Result<f64>

where T: PartialEq + Copy + Default,

Calculate F1 score

pub fn accuracy<T>(&self, predictions: &[T], truelabels: &[T]) -> Result<f64>

where T: PartialEq,

Calculate accuracy from predictions and true labels

pub fn binary_metrics<T>( &self, predictions: &[T], truelabels: &[T], ) -> Result<(f64, f64, f64)>

where T: PartialEq + Copy + Default + PartialEq<usize>,

Calculate precision, recall, and F1 score for binary classification

Trait Implementations

impl Clone for TextClassificationMetrics

fn clone(&self) -> TextClassificationMetrics

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TextClassificationMetrics

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for TextClassificationMetrics

fn default() -> Self

Returns the “default value” for a type.