probly.transformation.evidential.classification.torch¶
Torch evidential classification implementation.
Functions
Register a base model that the activation function will be appended to based on [SKK18]. |
- probly.transformation.evidential.classification.torch.append_activation_torch(obj)[source]¶
Register a base model that the activation function will be appended to based on [SKK18].
This applies a Softplus activation to ensure non-negative evidence, as required by Evidential Deep Learning.
- Parameters:
obj (Module)
- Return type: