@@ -383,15 +383,18 @@ class LightningLinkData(LightningDataModule):
383
383
Args:
384
384
data (Data or HeteroData): The :class:`~torch_geometric.data.Data` or
385
385
:class:`~torch_geometric.data.HeteroData` graph object.
386
- input_train_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor], optional ):
386
+ input_train_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor]):
387
387
The training edges. (default: :obj:`None`)
388
- input_train_edge_label (Tensor, optional): The labels of train edge indices.
389
- input_val_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor], optional):
388
+ input_train_edge_label (Tensor, optional):
389
+ The labels of train edge indices.
390
+ input_val_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor]):
390
391
The validation edges. (default: :obj:`None`)
391
- input_val_edge_label (Tensor, optional): The labels of val edge indices.
392
- input_test_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor], optional):
392
+ input_val_edge_label (Tensor, optional):
393
+ The labels of val edge indices.
394
+ input_test_edges (Tensor or EdgeType or Tuple[EdgeType, Tensor]):
393
395
The test edges. (default: :obj:`None`)
394
- input_test_edge_label (Tensor, optional): The labels of train edge indices.
396
+ input_test_edge_label (Tensor, optional):
397
+ The labels of train edge indices.
395
398
loader (str): The scalability technique to use (:obj:`"full"`,
396
399
:obj:`"link_neighbor"`). (default: :obj:`"link_neighbor"`)
397
400
batch_size (int, optional): How many samples per batch to load.
0 commit comments