models.emb.grarep

Module Contents

Classes

GraRep

The GraRep model from the `”Grarep: Learning graph representations with global structural information”

class models.emb.grarep.GraRep(dimension, step)[source]

Bases: models.BaseModel

The GraRep model from the “Grarep: Learning graph representations with global structural information” paper.

Args:

hidden_size (int) : The dimension of node representation. step (int) : The maximum order of transitition probability.

static add_args(parser)[source]

Add model-specific arguments to the parser.

classmethod build_model_from_args(cls, args)[source]

Build a new model instance.

train(self, G)[source]
_get_embedding(self, matrix, dimension)[source]