Enum Metric 

pub enum Metric {
    Mse,
    Mae,
    Accuracy,
    TopKAccuracy {
        k: usize,
    },
}

Supported evaluation metrics.

Variants

Mse

Mean squared error.

Mae

Mean absolute error.

Accuracy

Classification accuracy.

• For output_dim == 1: binary accuracy.
• For output_dim > 1: multiclass accuracy (argmax).

TopKAccuracy

Top-k accuracy for multiclass classification.

This metric requires output_dim > 1 and k <= output_dim.

Fields

k: usize

Implementations

impl Metric

pub fn validate(self) -> Result<()>

Validate metric parameters.

Trait Implementations

impl Clone for Metric

fn clone(&self) -> Metric

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Metric

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

Formats the value using the given formatter.

impl PartialEq for Metric

fn eq(&self, other: &Metric) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for Metric

impl Eq for Metric

impl StructuralPartialEq for Metric