Struct ConfusionMatrix 

pub struct ConfusionMatrix { /* private fields */ }

Confusion matrix for multi-class classification

Element [i][j] represents count of samples with true label i predicted as j

Implementations

impl ConfusionMatrix

pub fn new(n_classes: usize) -> Self

Create a new confusion matrix with given number of classes

pub fn from_predictions(y_pred: &[usize], y_true: &[usize]) -> Self

Create from predictions and ground truth

pub fn from_predictions_with_min_classes( y_pred: &[usize], y_true: &[usize], min_classes: usize, ) -> Self

Create from predictions and ground truth, ensuring at least min_classes classes

pub fn matrix(&self) -> &Vec<Vec<usize>>

Get the raw matrix

pub fn labels(&self) -> &[usize]

Get the class labels

pub fn n_classes(&self) -> usize

Get number of classes

pub fn get(&self, true_label: usize, predicted_label: usize) -> usize

Get element at [true_label][predicted_label]

pub fn true_positives(&self, class: usize) -> usize

Calculate true positives for a class

pub fn false_positives(&self, class: usize) -> usize

Calculate false positives for a class (predicted as class but wasn’t)

pub fn false_negatives(&self, class: usize) -> usize

Calculate false negatives for a class (was class but predicted differently)

pub fn true_negatives(&self, class: usize) -> usize

Calculate true negatives for a class

pub fn support(&self, class: usize) -> usize

Calculate support (total true instances) for a class

pub fn total(&self) -> usize

Total number of samples

pub fn accuracy(&self) -> f64

Calculate accuracy

Trait Implementations

impl Clone for ConfusionMatrix

fn clone(&self) -> ConfusionMatrix

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ConfusionMatrix

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

Formats the value using the given formatter.

impl Display for ConfusionMatrix

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

Formats the value using the given formatter.