torch_geometric.nn.models.ARGA

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

基础类：GAE

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

Parameters:

encoder (torch.nn.Module) – The encoder module.
discriminator (torch.nn.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().

Return type:: Tensor

reset_parameters()[source]: 重置模块的所有可学习参数。

reg_loss(z: Tensor) → Tensor[source]

计算编码器的正则化损失。

Parameters:: z (torch.Tensor) – The latent space \(\mathbf{Z}\).
Return type:: Tensor

discriminator_loss(z: Tensor) → Tensor[source]

计算判别器的损失。

Parameters:: z (torch.Tensor) – The latent space \(\mathbf{Z}\).
Return type:: Tensor