H2GB.encoder.Node2VecNodeEncoder
- class Node2VecNodeEncoder(dim_emb, dataset, reshape_x=True)[source]
Bases:
HeteroPENodeEncoder
Node2Vec based node encoder, based on the Node2Vec model from the “node2vec: Scalable Feature Learning for Networks” paper where random walks of length
walk_length
are sampled in a given graph, and node embeddings are learned via negative sampling optimization.