Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "85494e88189aa9aedf98f22ff6d61da39ebd2800"
Commit 18ed7868 authored by Da Zheng's avatar Da Zheng Committed by Minjie Wang
Browse files

[Doc] update neighbor sampler doc. (#287)

parent 597ac7f8
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
python/dgl/contrib/sampling/sampler.py
View file @ 18ed7868
...@@ -130,8 +130,9 @@ def NeighborSampler(g, batch_size, expand_factor, num_hops=1, ...@@ -130,8 +130,9 @@ def NeighborSampler(g, batch_size, expand_factor, num_hops=1,
Returns Returns
------- -------
A subgraph loader that returns a list of batched subgraphs and a dictionary of A subgraph iterator
additional information about the subgraphs. The iterator returns a list of batched subgraphs and a dictionary of additional
information about the subgraphs.
''' '''
return NSSubgraphLoader(g, batch_size, expand_factor, num_hops, neighbor_type, node_prob, return NSSubgraphLoader(g, batch_size, expand_factor, num_hops, neighbor_type, node_prob,
seed_nodes, shuffle, num_workers, max_subgraph_size, return_seed_id) seed_nodes, shuffle, num_workers, max_subgraph_size, return_seed_id)
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