pyraug.models

Here will be implemented the main Variational Autoencoders architecture you may use to perform data augmentation

By convention, each implemented model is contained within a folder located in pyraug.models in which are located 4 modules:

• model_config.py: Contains a OtherModelConfig instance inheriting from BaseModelConfig where the model configuration is stored and a OtherModelSamplerConfig instance inheriting from BaseSamplerConfig where the configuration of the sampler used to generate new samples is defined.
• other_model_model.py: An implementation of the other_model inheriting from BaseVAE.
• other_model_sampler.py: An implementation of the sampler(s) to use to generate new data inheriting from BaseSampler.

• other_model_utils.py: A module where utils methods are stored.

• BaseVAE
• RHVAE

BaseVAE	Base class for VAE based models.
RHVAE	This is an implementation of the Riemannian Hamiltonian VAE model proposed in (https://arxiv.org/pdf/2010.11518.pdf).