AbstractVariationalFamily#

class gpjax.variational_families.AbstractVariationalFamily(posterior)[source]#

Bases: _SummaryMixin, Module, Generic[L]

Abstract base class used to represent families of distributions that can be used within variational inference.

Parameters:

posterior (AbstractPosterior)

abstractmethod predict(*args, **kwargs)[source]#

Predict the GP’s output given the input.

Parameters:

  • *args (Any) – Arguments of the variational family’s predict method.

  • **kwargs (Any) – Keyword arguments of the variational family’s predict method.

Returns:

The output of the variational family’s predict method.

Return type:

GaussianDistribution