gensbi.experimental.recipes

Submodules

• gensbi.experimental.recipes.vae_pipeline

Classes

VAE1DPipeline	Pipeline for training and evaluating 1D Variational Autoencoders (VAE1D) in GenSBI.
VAE2DPipeline	Pipeline for training and evaluating 2D Variational Autoencoders (VAE2D) in GenSBI.

Package Contents

class gensbi.experimental.recipes.VAE1DPipeline(train_dataset, val_dataset, params, training_config=None)

Bases: AbstractVAEPipeline

Pipeline for training and evaluating 1D Variational Autoencoders (VAE1D) in GenSBI.

Inherits from AbstractVAEPipeline and uses the AutoEncoder1D model class.

Parameters:

params (gensbi.experimental.models.autoencoders.AutoEncoderParams)

class gensbi.experimental.recipes.VAE2DPipeline(train_dataset, val_dataset, params, training_config=None)

Bases: AbstractVAEPipeline

Pipeline for training and evaluating 2D Variational Autoencoders (VAE2D) in GenSBI.

Inherits from AbstractVAEPipeline and uses the AutoEncoder2D model class.

Parameters:

params (gensbi.experimental.models.autoencoders.AutoEncoderParams)