models.emb.prone

Module Contents

Classes

ProNE

The ProNE model from the `”ProNE: Fast and Scalable Network Representation Learning”

class models.emb.prone.ProNE(dimension, step, mu, theta)

Bases: models.BaseModel

The ProNE model from the “ProNE: Fast and Scalable Network Representation Learning” paper.

Args:

hidden_size (int) : The dimension of node representation. step (int) : The number of items in the chebyshev expansion. mu (float) : Parameter in ProNE. theta (float) : Parameter in ProNE.

static add_args(parser)

Add model-specific arguments to the parser.

classmethod build_model_from_args(cls, args)

Build a new model instance.

train(self, G)

_get_embedding_rand(self, matrix)

_get_embedding_dense(self, matrix, dimension)

_pre_factorization(self, tran, mask)

_chebyshev_gaussian(self, A, a, order=5, mu=0.5, s=0.2, plus=False, nn=False)