UNAGI.model.models

Classes

Discriminator(input_dim)

VAE(input_dim, hidden_dim, graph_dim, latent_dim)