probly.transformation.evidential.regression.torch¶
Torch evidential regression implementation.
Functions
|
Register a class to be replaced by the NormalInverseGammaLinear layer based on [ASSR20]. |
- probly.transformation.evidential.regression.torch.replace_last_torch_nig(obj, state)[source]¶
Register a class to be replaced by the NormalInverseGammaLinear layer based on [ASSR20].
This layer outputs the parameters of a Normal Inverse Gamma distribution, which is central to evidential regression.
- Parameters:
obj (nn.Linear)
state (State)
- Return type:
TraverserResult