dmlc / dmlc/dgl

Node sampling via probability vector

Open
#4,448 2 comments 0 reactions 1 assignee Claimed by @Rhett-Ying View on GitHub
feature request
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.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.