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):
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.npy
path: set/paper-validation-node_pairs.npy
- name: labels
format: numpy
in_memory: true
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
----------
......
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