Module

Neural network components for the iVAE architecture.

iVAE.module.weight_init	Initialize weights for linear layers using Xavier normal initialization.
iVAE.module.Encoder	Encoder network for the VAE that maps input data to latent space.
iVAE.module.Decoder	Decoder network for the VAE that reconstructs data from latent space.
iVAE.module.VAE	Interpretable Variational Autoencoder (iVAE) with an interpretative module.