gensbi.solver#

Abstract base class for solvers.

Classes#

Solver

Abstract base class for generative model solvers.

Module Contents#

class gensbi.solver.Solver[source]#

Bases: abc.ABC

Abstract base class for generative model solvers.

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

Compute the log-probability for given samples.

Only supported by solvers that can evaluate the continuous change-of-variables formula (e.g. ODESolver).

Raises:

NotImplementedError – If the solver does not support log-probability computation.

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

Return a callable that computes the log-probability.

Only supported by solvers that can evaluate the continuous change-of-variables formula (e.g. ODESolver).

Raises:

NotImplementedError – If the solver does not support log-probability computation.

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

Sample from the solver.

Parameters:

  • *args (Any) – Positional arguments.

  • **kwargs (Any) – Keyword arguments.

Returns:

Sampled output from the solver.

Return type:

Array