torch_geometric.nn.models.ARGVA
- class ARGVA(encoder: Module, discriminator: Module, decoder: Optional[Module] = None)[source]
Bases:
ARGA
The Adversarially Regularized Variational Graph Auto-Encoder model from the “Adversarially Regularized Graph Autoencoder for Graph Embedding” paper.
- Parameters:
encoder (torch.nn.Module) – The encoder module to compute \(\mu\) and \(\log\sigma^2\).
discriminator (torch.nn.Module) – The discriminator module.
decoder (torch.nn.Module, optional) – The decoder module. If set to
None
, will default to thetorch_geometric.nn.models.InnerProductDecoder
. (default:None
)
- reset_parameters()
Resets all learnable parameters of the module.