Algorithms

Node2Vec(edge_index, embedding_dim, ...[, ...])

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.