cugraph_pyg.loader.link_neighbor_loader.LinkNeighborLoader#
- class cugraph_pyg.loader.link_neighbor_loader.LinkNeighborLoader(
- data: torch_geometric.data.Data | torch_geometric.data.HeteroData | Tuple[torch_geometric.data.FeatureStore, torch_geometric.data.GraphStore],
- num_neighbors: List[int] | Dict[torch_geometric.typing.EdgeType, List[int]],
- edge_label_index: torch_geometric.typing.InputEdges = None,
- edge_label: torch_geometric.typing.OptTensor = None,
- edge_label_time: torch_geometric.typing.OptTensor = None,
- replace: bool = False,
- subgraph_type: torch_geometric.typing.SubgraphType | str = 'directional',
- disjoint: bool = False,
- temporal_strategy: str = 'uniform',
- neg_sampling: torch_geometric.sampler.NegativeSampling | None = None,
- neg_sampling_ratio: int | float | None = None,
- time_attr: str | None = None,
- weight_attr: str | None = None,
- transform: Callable | None = None,
- transform_sampler_output: Callable | None = None,
- is_sorted: bool = False,
- filter_per_worker: bool | None = None,
- neighbor_sampler: torch_geometric.sampler.NeighborSampler | None = None,
- directed: bool = True,
- batch_size: int = 16,
- compression: str | None = None,
- local_seeds_per_call: int | None = None,
- temporal_comparison: str | None = None,
- **kwargs,
Duck-typed version of torch_geometric.loader.LinkNeighborLoader
Link loader that implements the neighbor sampling algorithm used in GraphSAGE.
- __init__(
- data: torch_geometric.data.Data | torch_geometric.data.HeteroData | Tuple[torch_geometric.data.FeatureStore, torch_geometric.data.GraphStore],
- num_neighbors: List[int] | Dict[torch_geometric.typing.EdgeType, List[int]],
- edge_label_index: torch_geometric.typing.InputEdges = None,
- edge_label: torch_geometric.typing.OptTensor = None,
- edge_label_time: torch_geometric.typing.OptTensor = None,
- replace: bool = False,
- subgraph_type: torch_geometric.typing.SubgraphType | str = 'directional',
- disjoint: bool = False,
- temporal_strategy: str = 'uniform',
- neg_sampling: torch_geometric.sampler.NegativeSampling | None = None,
- neg_sampling_ratio: int | float | None = None,
- time_attr: str | None = None,
- weight_attr: str | None = None,
- transform: Callable | None = None,
- transform_sampler_output: Callable | None = None,
- is_sorted: bool = False,
- filter_per_worker: bool | None = None,
- neighbor_sampler: torch_geometric.sampler.NeighborSampler | None = None,
- directed: bool = True,
- batch_size: int = 16,
- compression: str | None = None,
- local_seeds_per_call: int | None = None,
- temporal_comparison: str | None = None,
- **kwargs,
- data: Data, HeteroData, or Tuple[FeatureStore, GraphStore]
See torch_geometric.loader.LinkNeighborLoader.
- num_neighbors: List[int] or Dict[EdgeType, List[int]]
Fanout values. See torch_geometric.loader.LinkNeighborLoader.
- edge_label_index: InputEdges
Input edges for sampling. See torch_geometric.loader.LinkNeighborLoader.
- edge_label: OptTensor
Labels for input edges. See torch_geometric.loader.LinkNeighborLoader.
- edge_label_time: OptTensor
Time attribute for input edges. See torch_geometric.loader.LinkNeighborLoader.
- replace: bool (optional, default=False)
Whether to sample with replacement. See torch_geometric.loader.LinkNeighborLoader.
- subgraph_type: Union[SubgraphType, str] (optional, default=’directional’)
The type of subgraph to return. Currently only ‘directional’ is supported. See torch_geometric.loader.LinkNeighborLoader.
- disjoint: bool (optional, default=False)
Whether to perform disjoint sampling. See torch_geometric.loader.LinkNeighborLoader.
- temporal_strategy: str (optional, default=’uniform’)
Currently only ‘uniform’ is suppported. See torch_geometric.loader.LinkNeighborLoader.
- time_attr: str (optional, default=None)
Used for temporal sampling. See torch_geometric.loader.LinkNeighborLoader.
- weight_attr: str (optional, default=None)
Used for biased sampling. See torch_geometric.loader.LinkNeighborLoader.
- transform: Callable (optional, default=None)
See torch_geometric.loader.LinkNeighborLoader.
- transform_sampler_output: Callable (optional, default=None)
See torch_geometric.loader.LinkNeighborLoader.
- is_sorted: bool (optional, default=False)
Ignored by cuGraph. See torch_geometric.loader.LinkNeighborLoader.
- filter_per_worker: bool (optional, default=False)
Currently ignored by cuGraph, but this may change once in-memory sampling is implemented. See torch_geometric.loader.LinkNeighborLoader.
- neighbor_sampler: torch_geometric.sampler.NeighborSampler
(optional, default=None) Not supported by cuGraph. See torch_geometric.loader.LinkNeighborLoader.
- directed: bool (optional, default=True)
Deprecated. See torch_geometric.loader.LinkNeighborLoader.
- batch_size: int (optional, default=16)
The number of input nodes per output minibatch. See torch.utils.dataloader.
- compression: str (optional, default=None)
The compression type to use if writing samples to disk. If not provided, it is automatically chosen.
- local_seeds_per_call: int (optional, default=None)
The number of seeds to process within a single sampling call. Manually tuning this parameter is not recommended but reducing it may conserve GPU memory. The total number of seeds processed per sampling call is equal to the sum of this parameter across all workers. If not provided, it will be automatically calculated. See cugraph_pyg.sampler.BaseDistributedSampler.
- temporal_comparison: str (optional, default=’monotonically_decreasing’)
The comparison operator for temporal sampling (‘strictly_increasing’, ‘monotonically_increasing’, ‘strictly_decreasing’, ‘monotonically_decreasing’, ‘last’). Note that this should be ‘last’ for temporal_strategy=’last’. See cugraph_pyg.sampler.BaseDistributedSampler.
- **kwargs
Other keyword arguments passed to the superclass.
Methods
__init__(data, num_neighbors[, ...])data: Data, HeteroData, or Tuple[FeatureStore, GraphStore]