UNAGI.train.customized_elbo

The customized elbo to support the VAE and discriminator loss.

Functions

graphUpdater(loss, model, guide, ...[, ...])

updater of Graph VAE-GAN.

Classes

myELBO([geneWeight, pushback_Score])

The customized ELBO function for the VAE-GAN model.