EmbeddingMixin

class EmbeddingMixin[source]

Bases: object

__init__()

Methods

__init__

embed

Compute embedding for model and data
embed(data)[source]

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