probly.train.evidential.torch.predictive_probs

probly.train.evidential.torch.predictive_probs(alpha: Tensor) → Tensor

Expected categorical probabilities under Dirichlet.

Used by Posterior Networks, Dirichlet Prior Networks, and other Dirichlet-based classification models to obtain predictive class probabilities.

Parameters:

alpha – Dirichlet concentration parameters, shape (B, C).

Returns:

Expected categorical probabilities, shape (B, C).