Enum SupervisedLoss 

pub enum SupervisedLoss {
    Mse,
    Mae,
    Huber {
        delta: f32,
    },
    Hinge {
        margin: f32,
    },
    Bce,
    Nll,
    CrossEntropy,
}

Configures supervised-loss function used by train-step and train-epoch helpers.

Variants

Mse

Mae

Huber

Fields

delta: f32

Hinge

Fields

margin: f32

Bce

Nll

CrossEntropy

Trait Implementations

impl Clone for SupervisedLoss

fn clone(&self) -> SupervisedLoss

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SupervisedLoss

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

Formats the value using the given formatter.

impl Default for SupervisedLoss

fn default() -> SupervisedLoss

Returns the “default value” for a type.

impl PartialEq for SupervisedLoss

fn eq(&self, other: &SupervisedLoss) -> 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 SupervisedLoss

impl StructuralPartialEq for SupervisedLoss