DistributionNLLLoss

class torch_uncertainty.losses.DistributionNLLLoss(reduction='mean')

Negative Log-Likelihood loss using given distributions as inputs.

Parameters:

• reduction (str, optional) – specifies the reduction to apply to the

• output – 'none' | 'mean' | 'sum'. Defaults to “mean”.

forward(dist, targets)

Compute the NLL of the targets given predicted distributions.

Parameters:

• dist (Distribution) – The predicted distributions

• targets (Tensor) – The target values