gensbi.models.wrappers.unconditional#

Unconditional model wrapper for GenSBI.

This module provides a wrapper class for unconditional models used in flow matching, handling proper input expansion and calling conventions.

Classes#

UnconditionalWrapper

Wrapper for unconditional models to handle input expansion and calling convention.

Module Contents#

class gensbi.models.wrappers.unconditional.UnconditionalWrapper(model)[source]#

Bases: gensbi.utils.model_wrapping.ModelWrapper

Wrapper for unconditional models to handle input expansion and calling convention.

Parameters:

model (The unconditional model instance to wrap.)

__call__(t, obs, obs_ids, **kwargs)[source]#

Call the wrapped model with expanded inputs.

Parameters:

  • t (Array) – Time steps.

  • obs (Array) – Observations.

  • obs_ids (Array) – Observation identifiers.

  • **kwargs (Additional keyword arguments passed to the model.)

Returns:

Model output.

Return type:

Array