malt.models.regressor.HomoschedasticGaussianLikelihood

class malt.models.regressor.HomoschedasticGaussianLikelihood[source]

Bases: malt.models.regressor.NeuralNetworkLikelihood

Models a homoschedastic Gaussian likelihood to be used with neural network regressors. (Admits unconstrained parameters.)

Variables

2 (in_features =) –

Parameters

  • mu (torch.Tensor)

  • log_sigma (torch.Tensor)

Examples

>>> likelihood = HomoschedasticGaussianLikelihood()
>>> posterior = likelihood(torch.tensor(0.0))
__init__()

Methods

__init__()

Attributes

in_features