Skip to content

GitLab

Unverified Commit 1502c56d authored by Hongyu Cai's avatar Hongyu Cai Committed by GitHub
Browse files

[Doc] Fix typo of datasets (#2517) 


Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 1e69bf53
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
examples/pytorch/ogb/cluster-gat/main.py
View file @ 1502c56d
......@@ -232,7 +232,7 @@ if __name__ == '__main__':
else:
device = th.device('cpu')
# load reddit data
# load ogbn-products data
data = DglNodePropPredDataset(name='ogbn-products')
splitted_idx = data.get_idx_split()
train_idx, val_idx, test_idx = splitted_idx['train'], splitted_idx['valid'], splitted_idx['test']
......
examples/pytorch/ogb/cluster-sage/main.py
View file @ 1502c56d
......@@ -202,7 +202,7 @@ if __name__ == '__main__':
else:
device = th.device('cpu')
# load reddit data
# load ogbn-products data
data = DglNodePropPredDataset(name='ogbn-products')
splitted_idx = data.get_idx_split()
train_idx, val_idx, test_idx = splitted_idx['train'], splitted_idx['valid'], splitted_idx['test']
......
examples/pytorch/ogb/ogbn-products/graphsage/main.py
View file @ 1502c56d
......@@ -214,7 +214,7 @@ if __name__ == '__main__':
else:
device = th.device('cpu')
# load reddit data
# load ogbn-products data
data = DglNodePropPredDataset(name='ogbn-products')
splitted_idx = data.get_idx_split()
train_idx, val_idx, test_idx = splitted_idx['train'], splitted_idx['valid'], splitted_idx['test']
......
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