Unverified Commit 65272a53 authored by Xinyu Yao's avatar Xinyu Yao Committed by GitHub
Browse files

[GraphBolt] Update docstring of `ondisk dataset`. (#7306)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
parent 7612af0f
...@@ -636,28 +636,28 @@ class OnDiskDataset(Dataset): ...@@ -636,28 +636,28 @@ class OnDiskDataset(Dataset):
train_set: train_set:
- type: paper # could be omitted for homogeneous graph. - type: paper # could be omitted for homogeneous graph.
data: # multiple data sources could be specified. data: # multiple data sources could be specified.
- name: node_pairs - name: seeds
format: numpy # Can be numpy or torch. format: numpy # Can be numpy or torch.
in_memory: true # If not specified, default to true. in_memory: true # If not specified, default to true.
path: set/paper-train-node_pairs.npy path: set/paper-train-seeds.npy
- name: labels - name: labels
format: numpy format: numpy
path: set/paper-train-labels.npy path: set/paper-train-labels.npy
validation_set: validation_set:
- type: paper - type: paper
data: data:
- name: node_pairs - name: seeds
format: numpy format: numpy
path: set/paper-validation-node_pairs.npy path: set/paper-validation-seeds.npy
- name: labels - name: labels
format: numpy format: numpy
path: set/paper-validation-labels.npy path: set/paper-validation-labels.npy
test_set: test_set:
- type: paper - type: paper
data: data:
- name: node_pairs - name: seeds
format: numpy format: numpy
path: set/paper-test-node_pairs.npy path: set/paper-test-seeds.npy
- name: labels - name: labels
format: numpy format: numpy
path: set/paper-test-labels.npy path: set/paper-test-labels.npy
......
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