probly.method.evidential.classification.common.EvidentialClassificationPredictor

class probly.method.evidential.classification.common.EvidentialClassificationPredictor(*args, **kwargs)[source]

Bases: DirichletDistributionPredictor, Protocol, Generic[In, Out]

Protocol for Sensoy-like evidential classification predictors.

The network outputs Dirichlet concentration parameters alpha directly, computed internally as alpha = softplus(base(x)) + 1.