DECLoss#
- class torch_uncertainty.losses.DECLoss(annealing_step=None, reg_weight=None, loss_type='log', reduction='mean')[source]#
The deep evidential classification loss.
- Parameters:
annealing_step (int) – Annealing step for the weight of the regularization term.
reg_weight (float) – Fixed weight of the regularization term.
loss_type (str, optional) – Specifies the loss type to apply to the Dirichlet parameters:
'mse'
|'log'
|'digamma'
.reduction (str, optional) – Specifies the reduction to apply to the output:
'none'
|'mean'
|'sum'
.
References