Skip to content

[FEA] Support Time Filtering in Hop 0 During Temporal Sampling #5263

@alexbarghi-nv

Description

@alexbarghi-nv

Currently, hop 0 vertices are allowed to select any edge during sampling. However, this conflicts with the PyG API, especially when performing edge-based sampling, since the API allows for overrides in hop 0. Without support for hop 0 filtering, the accuracy of edge-based sampling in cuGraph-PyG will probably worse than the accuracy in native PyG.

Supporting this involves the following changes to the API:

  1. Allow specifying a "time override" for each seed in hop 0
  2. When a time override is specified, use those times as the constraints for edge selection in hop 0. After hop 0, the overrides should be ignored.

Metadata

Metadata

Labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions