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:
forward(*args, **kwargs) Tensor

Alias for encode().

Return type:

Tensor

reset_parameters()

Resets all learnable parameters of the module.