Struct ConfusionMatrix 

pub struct ConfusionMatrix {
    pub matrix: Vec<Vec<usize>>,
    pub n_classes: usize,
}

Simple confusion matrix for evaluation.

Fields

matrix: Vec<Vec<usize>>

The matrix values (row = true, col = pred).

n_classes: usize

Number of classes.

Implementations

impl ConfusionMatrix

pub fn from_predictions( predictions: &[usize], targets: &[usize], n_classes: usize, ) -> Self

Create a new confusion matrix from predictions and targets.

pub fn accuracy(&self) -> f32

Get accuracy.

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

Get precision for a class.

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

Get recall for a class.

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

Get F1 score for a class.

pub fn macro_f1(&self) -> f32

Get macro-averaged F1 score.

pub fn to_table(&self) -> String

Get a formatted string representation.

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<'de> Deserialize<'de> for ConfusionMatrix

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for ConfusionMatrix

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.