rusty1s / pytorch_cluster

PyTorch Extension Library of Optimized Graph Cluster Algorithms

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Implement weighted random walks

pbielak opened this issue · comments

In the Node2vec paper, the transition probabilities are computed based on both edge weights and the hyperparameters p and q. The current implementation does not include the edge weights of a graph. This feature was already requested in #88 and #115. Based on the rejection sampling method used for the current implementation [1], I implemented a version that includes edge weights.

The PR for this feature implementation is #140

[1] https://louisabraham.github.io/articles/node2vec-sampling.html

This issue had no activity for 6 months. It will be closed in 2 weeks unless there is some new activity. Is this issue already resolved?