Struct ConfusionMatrix 

pub struct ConfusionMatrix {
    pub predicted_totals: HashMap<String, usize>,
    pub actual_totals: HashMap<String, usize>,
    /* private fields */
}

Confusion matrix for entity type predictions.

Shows which entity types get confused with which others.

Fields

predicted_totals: HashMap<String, usize>

Total predictions per type

actual_totals: HashMap<String, usize>

Total ground truth per type

Implementations

impl ConfusionMatrix

pub fn new() -> Self

Create a new empty confusion matrix.

pub fn add(&mut self, predicted: &str, actual: &str)

Add a prediction-actual pair.

pub fn get(&self, predicted: &str, actual: &str) -> usize

Get count for a prediction-actual pair.

pub fn types(&self) -> Vec<String>

Get all entity types in the matrix.

pub fn precision(&self, entity_type: &str) -> f64

Get per-type precision.

pub fn recall(&self, entity_type: &str) -> f64

Get per-type recall.

pub fn most_confused(&self, top_n: usize) -> Vec<(String, String, usize)>

Get most confused pairs (predicted, actual, count).

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 Default for ConfusionMatrix

fn default() -> ConfusionMatrix

Returns the “default value” for a type.

impl Display for ConfusionMatrix

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

Formats the value using the given formatter.