Unverified Commit 50b05723 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[GraphBolt] update names of ItemSet in OnDiskDataset testcases (#6289)

parent 19d63943
...@@ -313,24 +313,36 @@ class OnDiskDataset(Dataset): ...@@ -313,24 +313,36 @@ class OnDiskDataset(Dataset):
train_set: train_set:
- type: paper # could be null for homogeneous graph. - type: paper # could be null for homogeneous graph.
data: # multiple data sources could be specified. data: # multiple data sources could be specified.
- format: numpy - name: node_pairs
format: numpy
in_memory: true # If not specified, default to true. in_memory: true # If not specified, default to true.
path: set/paper-train-src.npy path: set/paper-train-node_pairs.npy
- format: numpy - name: labels
format: numpy
in_memory: false in_memory: false
path: set/paper-train-dst.npy path: set/paper-train-labels.npy
validation_set: validation_set:
- type: paper - type: paper
data: 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 in_memory: true
path: set/paper-validation.npy path: set/paper-validation-labels.npy
test_set: test_set:
- type: paper - type: paper
data: 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 in_memory: true
path: set/paper-test.npy path: set/paper-test-labels.npy
Parameters Parameters
---------- ----------
......
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