Skip to content

GitLab

Unverified Commit 073a270e authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[GraphBolt] update example to use built-in dataset (#6336)

parent fe96f11f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
examples/sampling/graphbolt/link_prediction.py
View file @ 073a270e
...@@ -328,9 +328,7 @@ def main(args): ...@@ -328,9 +328,7 @@ def main(args):
# Load and preprocess dataset. # Load and preprocess dataset.
print("Loading data") print("Loading data")
dataset = gb.OnDiskDataset( dataset = gb.BuiltinDataset("ogbl-citation2").load()
"/home/ubuntu/wklwork/work#4/example_ogbl_citation2"
).load()
graph = dataset.graph graph = dataset.graph
features = dataset.feature features = dataset.feature
train_set = dataset.tasks[0].train_set train_set = dataset.tasks[0].train_set
......
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