Skip to content

Add DistPPRSampler#538

Merged
mkolodner-sc merged 116 commits intomainfrom
mkolodner-sc/add_ppr_neighbor_sampler
Mar 21, 2026
Merged

Add DistPPRSampler#538
mkolodner-sc merged 116 commits intomainfrom
mkolodner-sc/add_ppr_neighbor_sampler

Conversation

@mkolodner-sc
Copy link
Collaborator

@mkolodner-sc mkolodner-sc commented Mar 9, 2026


Summary

Adds a Personalized PageRank (PPR) neighbor sampler (DistPPRNeighborSampler) as an alternative to the default k-hop uniform neighbor sampler.

  • New DistPPRNeighborSampler — implements the Forward Push algorithm (Andersen et al., 2006) with support for both homogeneous and heterogeneous graphs, including ABLP inputs
  • New PPRSamplerOptions — frozen dataclass exposing alpha, eps, max_ppr_nodes, num_nbrs_per_hop, and total_degree_dtype, plumbed through DistSamplingProducer so it works in both colocated and graph store modes
  • Comprehensive tests — PPR correctness validated against NetworkX PageRank for both homogeneous and heterogeneous graphs, parameterized over edge_dir, with ABLP metadata propagation tests

mkolodner-sc and others added 26 commits March 4, 2026 22:07
…pler_cls; make sampler_options required on dist_server
@mkolodner-sc mkolodner-sc changed the base branch from main to mkolodner-sc/fix_test_dataset_edge_dir_handling March 20, 2026 01:10
Base automatically changed from mkolodner-sc/fix_test_dataset_edge_dir_handling to main March 20, 2026 23:25
Copy link
Collaborator

@yliu2-sc yliu2-sc left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the work!

@mkolodner-sc mkolodner-sc enabled auto-merge March 20, 2026 23:36
@mkolodner-sc mkolodner-sc added this pull request to the merge queue Mar 20, 2026
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Mar 21, 2026
@mkolodner-sc mkolodner-sc added this pull request to the merge queue Mar 21, 2026
Merged via the queue into main with commit e2f2180 Mar 21, 2026
6 checks passed
@mkolodner-sc mkolodner-sc deleted the mkolodner-sc/add_ppr_neighbor_sampler branch March 21, 2026 02:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants