Skip to content

GitLab

Unverified Commit 11e910b6 authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

[Doc] Hotfix for GraphSAINT Doc (#3898) 

* Update graphsaint.py

* Update graphsaint.py
parent 35e66f42
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
python/dgl/dataloading/graphsaint.py
View file @ 11e910b6
...@@ -61,8 +61,7 @@ class SAINTSampler(Sampler): ...@@ -61,8 +61,7 @@ class SAINTSampler(Sampler):
>>> num_iters = 1000 >>> num_iters = 1000
>>> sampler = SAINTSampler(mode='node', budget=6000) >>> sampler = SAINTSampler(mode='node', budget=6000)
>>> # Assume g.ndata['feat'] and g.ndata['label'] hold node features and labels >>> # Assume g.ndata['feat'] and g.ndata['label'] hold node features and labels
>>> dataloader = DataLoader(g, torch.arange(num_iters), sampler, num_workers=4, >>> dataloader = DataLoader(g, torch.arange(num_iters), sampler, num_workers=4)
... prefetch_ndata=['feat', 'label'])
>>> for subg in dataloader: >>> for subg in dataloader:
... train_on(subg) ... train_on(subg)
""" """
......
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