Unverified Commit a548805d authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Rename the variables in NeighborSampler for better readability. (#6451)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent d1ee3a55
......@@ -113,20 +113,21 @@ class NeighborSampler(SubgraphSampler):
self.replace,
self.prob_name,
)
original_column_node_ids = seeds
if self.deduplicate:
seeds, compacted_node_pairs = unique_and_compact_node_pairs(
subgraph.node_pairs, seeds
)
(
original_row_node_ids,
compacted_node_pairs,
) = unique_and_compact_node_pairs(subgraph.node_pairs, seeds)
else:
raise RuntimeError("Not implemented yet.")
subgraph = SampledSubgraphImpl(
node_pairs=compacted_node_pairs,
original_column_node_ids=original_column_node_ids,
original_row_node_ids=seeds,
original_column_node_ids=seeds,
original_row_node_ids=original_row_node_ids,
original_edge_ids=subgraph.original_edge_ids,
)
subgraphs.insert(0, subgraph)
seeds = original_row_node_ids
return seeds, subgraphs
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment