The model architecture of UNAGI

The module contains the VAE model and the discriminator model.

UNAGI.model.models

Trainer of UNAGI

The modules to start the training of UNAGI.

UNAGI.train.trainer.UNAGI_trainer
UNAGI.train.runner.UNAGI_runner	The UNAGI_runner class is used to set up the hyperparameters to run iDREM, find clustering optimal parameters and run the UNAGI model .
UNAGI.train.customized_elbo	The customized elbo to support the VAE and discriminator loss.