Node sampling via probability vector
- Dominant language
- Python
- Stars
- 14.3k
- Forks
- 3.1k
- PR merge metrics
- No merged PRs in 30d
Description
## 🚀 Feature
A modification to the dgl.sampling.sample_neighbours() function that allows you to input a vector of transition probabilities between nodes (rather than having to add them as edge features). This would allow you to sample a graph based that isn't necessarily fully connected.
## Motivation
The downstream motivation is an implementation of LADIES. This mini-batch training approach uses probabilistic layer based sampling to generate an efficient and connected subgraph that reduces the memory and time complexity of training.
## Alternatives
Manually constructing a subgraph using the node ids. But this is kind of messy and inconvenient.
## Pitch
Implement a sampling method which samples based on a probability vector between nodes.
## Additional context
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.