search

shenweichen / GraphEmbedding

Implementation and experiments of graph embedding algorithms.
MIT License
3.63k stars 990 forks source link

KeyError during node2vec #40

Closed radkoff closed 3 years ago

radkoff commented 3 years ago

This fixes a KeyError seen during node2vec when using non-directed graphs (example here) by allowing the walker to initialize both directions of graph edges.