probly.method.evidential.regression.torch.replace_last_torch_nig¶
- probly.method.evidential.regression.torch.replace_last_torch_nig(obj: nn.Linear, state: State) TraverserResult[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.