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): ...@@ -113,20 +113,21 @@ class NeighborSampler(SubgraphSampler):
self.replace, self.replace,
self.prob_name, self.prob_name,
) )
original_column_node_ids = seeds
if self.deduplicate: 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: else:
raise RuntimeError("Not implemented yet.") raise RuntimeError("Not implemented yet.")
subgraph = SampledSubgraphImpl( subgraph = SampledSubgraphImpl(
node_pairs=compacted_node_pairs, node_pairs=compacted_node_pairs,
original_column_node_ids=original_column_node_ids, original_column_node_ids=seeds,
original_row_node_ids=seeds, original_row_node_ids=original_row_node_ids,
original_edge_ids=subgraph.original_edge_ids, original_edge_ids=subgraph.original_edge_ids,
) )
subgraphs.insert(0, subgraph) subgraphs.insert(0, subgraph)
seeds = original_row_node_ids
return seeds, subgraphs 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