torch_geometric.nn.models.ARGVA

class ARGVA(encoder: Module, discriminator: Module, decoder: Optional[Module] = None)[source]

基础类：ARGA

来自“对抗性正则化图自编码器用于图嵌入”论文的对抗性正则化变分图自编码器模型。

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().