l2gv2.embedding.gae.utils package

l2gv2.embedding.gae.utils package#

Submodules#

l2gv2.embedding.gae.utils.loss module#

TODO: module docstring for embedding/gae/utils/loss.py.

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

loss function for use with GAE_model()

Parameters:

  • model

  • data

Returns:

reconstruction loss

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

loss function for use with VGAE_model()

Parameters:

  • model

  • data

Returns:

reconstruction loss

l2gv2.embedding.gae.utils.mixins module#

TODO: module docstring for embedding gae/utils/mixins.py.

class l2gv2.embedding.gae.utils.mixins.EmbeddingMixin#

Bases: object

TODO: class docstring for EmbeddingMixin.

embed(data)#

Compute embedding for model and data

Parameters:

data – network

Returns:

embedding coords for nodes

This function switches the model to eval state before computing the embedding and restores the original training state of the model

Module contents#

Contents