Struct SmoothL1Loss

pub struct SmoothL1Loss {
    pub beta: f32,
    pub reduction: LossReduction,
}

Computes the smooth L1 loss.

The smooth L1 loss is a variant of the L1 loss which replaces the absolute difference with a squared difference when the absolute difference is less than beta.

Fields

beta: f32

The threshold after which the loss changes from the squared to the absolute difference. Default to SmoothL1Loss::DEFAULT_BETA

reduction: LossReduction

Reduction type. Default to SmoothL1Loss::DEFAULT_REDUCTION

Implementations

impl SmoothL1Loss

pub fn new() -> Self

Creates a new instance of SmoothL1Loss.

impl SmoothL1Loss

pub const DEFAULT_BETA: f32 = 1f32

Default value for the beta parameter.

pub const DEFAULT_REDUCTION: LossReduction = LossReduction::Mean

Default value for the reduction parameter.

pub fn apply( &self, predictions: impl AsRef<Array>, targets: impl AsRef<Array>, ) -> Result<Array, Exception>

Compute the smooth L1 loss.

Params

• predictions: predicted values
• targets: target values

Trait Implementations

impl Buildable for SmoothL1Loss

type Builder = SmoothL1LossBuilder

The builder type for this buildable type

impl Builder<SmoothL1Loss> for SmoothL1LossBuilder

type Error = Infallible

Error with building

fn build(self) -> Result<SmoothL1Loss, Self::Error>

Build the type

impl Clone for SmoothL1Loss

fn clone(&self) -> SmoothL1Loss

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for SmoothL1Loss

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

Formats the value using the given formatter.