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
47898ab6
Unverified
Commit
47898ab6
authored
Dec 01, 2023
by
Rhett Ying
Committed by
GitHub
Dec 01, 2023
Browse files
[doc] add docstring for SubgraphSampler::_sample_subgraphs() (#6655)
parent
0a3f6116
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
16 additions
and
0 deletions
+16
-0
python/dgl/graphbolt/subgraph_sampler.py
python/dgl/graphbolt/subgraph_sampler.py
+16
-0
No files found.
python/dgl/graphbolt/subgraph_sampler.py
View file @
47898ab6
...
@@ -135,4 +135,20 @@ class SubgraphSampler(MiniBatchTransformer):
...
@@ -135,4 +135,20 @@ class SubgraphSampler(MiniBatchTransformer):
)
)
def
_sample_subgraphs
(
self
,
seeds
):
def
_sample_subgraphs
(
self
,
seeds
):
"""Sample subgraphs from the given seeds.
Any subclass of SubgraphSampler should implement this method.
Parameters
----------
seeds : Union[torch.Tensor, Dict[str, torch.Tensor]]
The seed nodes.
Returns
-------
Union[torch.Tensor, Dict[str, torch.Tensor]]
The input nodes.
SampledSubgraph
The sampled subgraphs.
"""
raise
NotImplementedError
raise
NotImplementedError
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