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, **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, **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. Currently unsupported. 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.

**kwargs

Other keyword arguments passed to the superclass.

Methods

__init__(data, num_neighbors[, ...])

data: Data, HeteroData, or Tuple[FeatureStore, GraphStore]