Learn R Programming

DistributionIV (version 0.1.2)

energyloss: 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)

Value

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

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||).