We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ef4924e commit dc0511cCopy full SHA for dc0511c
gammagl/datasets/amazon.py
@@ -32,11 +32,14 @@ class Amazon(InMemoryDataset):
32
an :obj:`gammagl.data.Graph` object and returns a
33
transformed version. The data object will be transformed before
34
being saved to disk. (default: :obj:`None`)
35
- force_reload (bool, optional): Whether to re-process the dataset.
+ force_reload : bool, optional
36
+ Whether to re-process the dataset.
37
(default: :obj:`False`)
- train_per_class (int, optional): Number of training samples per class.
38
+ train_per_class : int, optional
39
+ Number of training samples per class.
40
(default: :obj:`20`)
- val_per_class (int, optional): Number of validation samples per class.
41
+ val_per_class : int, optional
42
+ Number of validation samples per class.
43
44
45
Stats:
0 commit comments