Getting started with UNAGI:
Classes
Discriminator(input_dim)
Discriminator
VAE(input_dim, hidden_dim, graph_dim, latent_dim)
VAE