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 the torch_geometric.nn.models.InnerProductDecoder. (default: None)

forward(*args, **kwargs) → Tensor

Alias for encode().

reset_parameters()

Resets all learnable parameters of the module.