UNAGI.model.models

Classes

Discriminator(input_dim)

GCNLayer(in_features, out_features[, use_bias])

Graph_encoder(input_dim, hidden_dim, ...)

Plain_VAE(input_dim, hidden_dim, graph_dim, ...)

Plain_encoder(input_dim, hidden_dim, ...)

VAE(input_dim, hidden_dim, graph_dim, latent_dim)

VAE_decoder(input_dim, hidden_dim, ...)