Skip to content

Commit 97c1735

Browse files
BowenYao18mfbalin
andauthored
Update examples/graphbolt/rgcn/download.py
Co-authored-by: Muhammed Fatih BALIN <[email protected]>
1 parent 19f18a5 commit 97c1735

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/graphbolt/rgcn/download.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -397,7 +397,7 @@ def split_data(label_path, set_dir, dataset_size):
397397

398398
total_samples = len(labels)
399399
train_end = int(0.6 * total_samples)
400-
validation_end = int(0.9 * total_samples)
400+
validation_end = int(0.8 * total_samples)
401401

402402
indices = np.arange(total_samples)
403403
train_indices = indices[:train_end]

0 commit comments

Comments
 (0)