Skip to content

Commit d8e5c2f

Browse files
committed
fix doc dataset split typo
1 parent 51388f5 commit d8e5c2f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dhg/utils/dataset_split.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,10 @@ def split_by_num(
4343
>>> train_num, val_num, test_num = 10, 2, 5
4444
>>> train_mask, val_mask, test_mask = split_by_num(num_v, v_label, train_num, val_num, test_num)
4545
>>> train_mask.sum(), val_mask.sum(), test_mask.sum()
46-
(tensor(90), tensor(6), tensor(15))
46+
(tensor(30), tensor(6), tensor(15))
4747
>>> train_mask, val_mask, test_mask = split_by_num(num_v, v_label, train_num, val_num)
4848
>>> train_mask.sum(), val_mask.sum(), test_mask.sum()
49-
(tensor(90), tensor(6), tensor(64))
49+
(tensor(30), tensor(6), tensor(64))
5050
"""
5151
if isinstance(v_label, list):
5252
v_label = np.array(v_label)

0 commit comments

Comments
 (0)