Bi-directional Sampling for NeighborSampler #10126
Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
zaristei
force-pushed
the
zaristei/bidirectional-sampler
branch
5 times, most recently
from
37688a8 to
7ba87d5
Compare
This was referenced Mar 25, 2025
|
looks good so far |
puririshi98
added a commit
that referenced
this pull request
Apr 2, 2025
While working on [this PR](#10126), I noticed that there are no sanity tests for the default NeighborSampler. Although NeigborLoader effectively tests a lot of this functionality, having lower level tests helps with ensuring consistency for each `sample` call of the sampler, without the batching logic of `NeighborLoader` getting involved. --------- Co-authored-by: Zack Aristei <[email protected]> Co-authored-by: Rishi Puri <[email protected]>
zaristei
force-pushed
the
zaristei/bidirectional-sampler
branch
2 times, most recently
from
fdf6450 to
c9eab32
Compare
zaristei
force-pushed
the
zaristei/bidirectional-sampler
branch
2 times, most recently
from
d2506c8 to
9adc7b9
Compare
|
closing since matthias said theres an existing simple way to do this. please jsut put the fix into my branch: |
puririshi98
requested changes
Apr 15, 2025
There was a problem hiding this comment.
LGTM but i will request that @akihironitta or @rusty1s review this
zaristei
force-pushed
the
zaristei/bidirectional-sampler
branch
2 times, most recently
from
e288d28 to
d99fb71
Compare
|
#10200 should be merged first |
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## master #10126 +/- ##
==========================================
- Coverage 86.11% 85.46% -0.65%
==========================================
Files 496 496
Lines 33655 34129 +474
==========================================
+ Hits 28981 29169 +188
- Misses 4674 4960 +286 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
…rectional-sampler
zaristei
force-pushed
the
zaristei/bidirectional-sampler
branch
from
e6b47d0 to
a2c6b11
Compare
# Conflicts: # test/sampler/test_sampler_base.py # torch_geometric/sampler/base.py
zaristei
force-pushed
the
zaristei/bidirectional-sampler
branch
from
889e316 to
f5333cf
Compare
added 13 commits
April 22, 2025 14:18
# Conflicts: # test/sampler/test_sampler_base.py
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR makes the following changes:
srctodstnode, and not just fromdsttosrcnode, which is the current default behavior.BidirectionalNeighborSampler, which samples nodes/edges in both upstream and downstream directions for a directed graph, as if it were an undirected graph.merge_withandcollatemethods to the SamplerOutput dataclass, allowing for different SamplerOutputs to be merged together. This also becomes useful in scenarios where the user may want to sample using multiple different unique sampling methods.This PR will remain in draft until
methods are also implemented on the corresponding Hetero graph classes, andunittests have been written to cover new behaviors.UPDATE: Hetero Bidirectional Sampler and SamplerOutput merge will be a separate PR.