dirichlet_exp_activation

probly.transformation.dirichlet_exp_activation(base: T) T[source]

Append an exp activation so logits become Dirichlet concentrations.

Parameters:

base – The base logit classifier to be wrapped.

Returns:

A predictor returning Dirichlet alpha.