Struct GaussianNllLoss

pub struct GaussianNllLoss {
    pub full: bool,
    pub eps: f32,
    pub reduction: LossReduction,
}

Compute the negative log likelihood loss for a Gaussian distribution.

Fields

full: bool

Whether to include the constant term in the loss calculation. Default to GaussianNllLoss::DEFAULT_FULL

eps: f32

Small positive constant for numerical stability. Default to GaussianNllLoss::DEFAULT_EPS

reduction: LossReduction

Reduction type. Default to GaussianNllLoss::DEFAULT_REDUCTION

Implementations

impl GaussianNllLoss

pub fn new() -> Self

Creates a new instance of GaussianNllLoss.

impl GaussianNllLoss

pub const DEFAULT_FULL: bool = false

Default value for the full parameter.

pub const DEFAULT_EPS: f32 = 9.99999997E-7f32

Default value for the eps parameter.

pub const DEFAULT_REDUCTION: LossReduction = LossReduction::None

Default value for the reduction parameter.

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

Compute the negative log likelihood loss for a Gaussian distribution.

Params

• inputs: The predicted expectation of the Gaussian distribution.
• targets: The target values (samples from the Gaussian distribution).
• vars: The predicted variance of the Gaussian distribution.

Trait Implementations

impl Buildable for GaussianNllLoss

type Builder = GaussianNllLossBuilder

The builder type for this buildable type

impl Builder<GaussianNllLoss> for GaussianNllLossBuilder

type Error = Infallible

Error with building

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

Build the type

impl Clone for GaussianNllLoss

fn clone(&self) -> GaussianNllLoss

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for GaussianNllLoss

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

Formats the value using the given formatter.