gensbi.diffusion.path.path#

Classes#

ProbPath

Helper class that provides a standard way to create an ABC using

Module Contents#

class gensbi.diffusion.path.path.ProbPath(scheduler)[source]#

Bases: abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

Parameters:

scheduler (Any)

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

Abstract method to sample from the probability path.

Returns:

Sample from the path.

Return type:

PathSample

Parameters:

  • args (Any)

  • kwargs (Any)

sample_prior(key, shape)[source]#

Sample from the prior distribution.

Parameters:

  • key (Array) – JAX random key.

  • shape (Any) – Shape of the samples to generate, should be (nsamples, ndim).

Returns:

Samples from the prior distribution, shape (nsamples, ndim).

Return type:

Array

property name: str[source]#

Returns the name of the scheduler.

Returns:

Scheduler name.

Return type:

str

scheduler[source]#