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
50b05723
Unverified
Commit
50b05723
authored
Sep 06, 2023
by
Rhett Ying
Committed by
GitHub
Sep 06, 2023
Browse files
[GraphBolt] update names of ItemSet in OnDiskDataset testcases (#6289)
parent
19d63943
Changes
2
Expand all
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
158 additions
and
157 deletions
+158
-157
python/dgl/graphbolt/impl/ondisk_dataset.py
python/dgl/graphbolt/impl/ondisk_dataset.py
+20
-8
tests/python/pytorch/graphbolt/impl/test_ondisk_dataset.py
tests/python/pytorch/graphbolt/impl/test_ondisk_dataset.py
+138
-149
No files found.
python/dgl/graphbolt/impl/ondisk_dataset.py
View file @
50b05723
...
...
@@ -313,24 +313,36 @@ class OnDiskDataset(Dataset):
train_set:
- type: paper # could be null for homogeneous graph.
data: # multiple data sources could be specified.
- format: numpy
- name: node_pairs
format: numpy
in_memory: true # If not specified, default to true.
path: set/paper-train-src.npy
- format: numpy
path: set/paper-train-node_pairs.npy
- name: labels
format: numpy
in_memory: false
path: set/paper-train-
dst
.npy
path: set/paper-train-
labels
.npy
validation_set:
- type: paper
data:
- format: numpy
- name: node_pairs
format: numpy
in_memory: true
path: set/paper-validation-node_pairs.npy
- name: labels
format: numpy
in_memory: true
path: set/paper-validation.npy
path: set/paper-validation
-labels
.npy
test_set:
- type: paper
data:
- format: numpy
- name: node_pairs
format: numpy
in_memory: true
path: set/paper-test-node_pairs.npy
- name: labels
format: numpy
in_memory: true
path: set/paper-test.npy
path: set/paper-test
-labels
.npy
Parameters
----------
...
...
tests/python/pytorch/graphbolt/impl/test_ondisk_dataset.py
View file @
50b05723
This diff is collapsed.
Click to expand it.
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