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
073a270e
Unverified
Commit
073a270e
authored
Sep 15, 2023
by
Rhett Ying
Committed by
GitHub
Sep 15, 2023
Browse files
[GraphBolt] update example to use built-in dataset (#6336)
parent
fe96f11f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
3 deletions
+1
-3
examples/sampling/graphbolt/link_prediction.py
examples/sampling/graphbolt/link_prediction.py
+1
-3
No files found.
examples/sampling/graphbolt/link_prediction.py
View file @
073a270e
...
...
@@ -328,9 +328,7 @@ def main(args):
# Load and preprocess dataset.
print
(
"Loading data"
)
dataset
=
gb
.
OnDiskDataset
(
"/home/ubuntu/wklwork/work#4/example_ogbl_citation2"
).
load
()
dataset
=
gb
.
BuiltinDataset
(
"ogbl-citation2"
).
load
()
graph
=
dataset
.
graph
features
=
dataset
.
feature
train_set
=
dataset
.
tasks
[
0
].
train_set
...
...
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