energyloss {DistributionIV}R Documentation

Energy Loss Calculation

Description

This function calculates the energy loss for given tensors. The loss is calculated as the mean of the L2 norms between x0 and x and between x0 and xp, subtracted by half the mean of the L2 norm between x and xp.

Usage

energyloss(x0, x, xp, verbose = FALSE)

Arguments

x0

A tensor representing the target values.

x

A tensor representing the model's stochastic predictions.

xp

A tensor representing another draw of the model's stochastic predictions.

verbose

A boolean indicating whether to return prediction loss s1 = E(||x0-x||) and variance loss s2 = E(||x-xp||).

Value

A vector containing the calculated energy loss, s1, and s2, or a scalar representing the calculated energy loss.


[Package DistributionIV version 0.1.0 Index]