Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
18ed7868
"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "8e4fd686e0b45ac2406677b08ac3dca79aa290b3"
Commit
18ed7868
authored
Dec 11, 2018
by
Da Zheng
Committed by
Minjie Wang
Dec 11, 2018
Browse files
[Doc] update neighbor sampler doc. (#287)
parent
597ac7f8
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
python/dgl/contrib/sampling/sampler.py
python/dgl/contrib/sampling/sampler.py
+3
-2
No files found.
python/dgl/contrib/sampling/sampler.py
View file @
18ed7868
...
...
@@ -130,8 +130,9 @@ def NeighborSampler(g, batch_size, expand_factor, num_hops=1,
Returns
-------
A subgraph loader that returns a list of batched subgraphs and a dictionary of
additional information about the subgraphs.
A subgraph iterator
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
,
seed_nodes
,
shuffle
,
num_workers
,
max_subgraph_size
,
return_seed_id
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment