probly.train.evidential.torch.evidential_log_loss¶
- probly.train.evidential.torch.evidential_log_loss(alphas: Tensor, targets: Tensor) Tensor[source]¶
Evidential Log Loss for classification uncertainty estimation.
Implements the evidential log loss proposed by Sensoy et al. (2018) for Evidential Deep Learning.
- Reference:
Sensoy et al., “Evidential Deep Learning to Quantify Classification Uncertainty”, NeurIPS 2018. https://arxiv.org/abs/1806.01768
- Parameters:
alphas – Dirichlet concentration parameters, shape (B, C).
targets – Ground-truth class labels, shape (B,).
- Returns:
Scalar evidential log loss averaged over the batch.