l2gv2.embedding.gae package

l2gv2.embedding.gae package#

Subpackages#

Module contents#

TODO: module docstring for embedding/gae/__init__.py.

class l2gv2.embedding.gae.GAE(dim, hidden_dim, num_features, dist=False)#

Bases: GAE

TODO: class docstring for GAE.

l2gv2.embedding.gae.GAE_loss(model: Module, data: Data)#

loss function for use with GAE_model()

Parameters:

  • model

  • data

Returns:

reconstruction loss

class l2gv2.embedding.gae.GAEconv(dim, num_node_features, hidden_dim=32, cached=True, bias=True, add_self_loops=True, normalize=True)#

Bases: Module

implements the convolution operator for use with tg.nn.GAE

forward(data)#

compute coordinates given data

class l2gv2.embedding.gae.VGAE(dim, hidden_dim, num_features, dist=False)#

Bases: VGAE

VGAE model.

l2gv2.embedding.gae.VGAE_loss(model: Module, data: Data)#

loss function for use with VGAE_model()

Parameters:

  • model

  • data

Returns:

reconstruction loss

class l2gv2.embedding.gae.VGAEconv(dim, num_node_features, hidden_dim=32, cached=True, bias=True, add_self_loops=True, normalize=True)#

Bases: Module

implements the convolution operator for use with torch_geometric.nn.VGAE

forward(data: Data)#

compute mean and variance given data

Parameters:

[type] (data) – input data

Returns:

mu, sigma

Contents