Skip to content

Commit dc0511c

Browse files
committed
update amazon
1 parent ef4924e commit dc0511c

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

gammagl/datasets/amazon.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,14 @@ class Amazon(InMemoryDataset):
3232
an :obj:`gammagl.data.Graph` object and returns a
3333
transformed version. The data object will be transformed before
3434
being saved to disk. (default: :obj:`None`)
35-
force_reload (bool, optional): Whether to re-process the dataset.
35+
force_reload : bool, optional
36+
Whether to re-process the dataset.
3637
(default: :obj:`False`)
37-
train_per_class (int, optional): Number of training samples per class.
38+
train_per_class : int, optional
39+
Number of training samples per class.
3840
(default: :obj:`20`)
39-
val_per_class (int, optional): Number of validation samples per class.
41+
val_per_class : int, optional
42+
Number of validation samples per class.
4043
(default: :obj:`20`)
4144
4245
Stats:

0 commit comments

Comments
 (0)