Struct juice::solver::confusion_matrix::ConfusionMatrix

pub struct ConfusionMatrix { /* private fields */ }

A ConfusionMatrix.

Implementations

impl ConfusionMatrix

pub fn new(num_classes: usize) -> ConfusionMatrix

Create a ConfusionMatrix that analyzes the prediction of num_classes classes.

pub fn add_sample(&mut self, prediction: usize, target: usize)

Add a sample by providing the expected target class and the prediction.

pub fn add_samples(&mut self, predictions: &[usize], targets: &[usize])

Add a batch of samples.

See add_sample.

pub fn get_predictions(&self, network_out: &mut SharedTensor<f32>) -> Vec<usize>

Get the predicted classes from the output of a network.

The prediction for each sample of the batch is found by determining which output value had the smallest loss.

pub fn set_capacity(&mut self, capacity: Option<usize>)

Set the capacity of the ConfusionMatrix

pub fn samples(&self) -> &VecDeque<Sample>

Return all collected samples.

pub fn accuracy(&self) -> Accuracy

Return the accuracy of the collected predictions.

Trait Implementations

impl Debug for ConfusionMatrix

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

Formats the value using the given formatter.