cugraph_pyg.sampler.sampler.BaseSampler#

class cugraph_pyg.sampler.sampler.BaseSampler(sampler: DistributedNeighborSampler, data: Tuple[FeatureStore, GraphStore], batch_size: int = 16)[source]#

Methods

sample_from_edges
sample_from_nodes

__init__(sampler: DistributedNeighborSampler, data: Tuple[FeatureStore, GraphStore], batch_size: int = 16)[source]#

Methods

`__init__`(sampler, data[, batch_size])
`sample_from_edges`(index, neg_sampling, **kwargs)
`sample_from_nodes`(index, **kwargs)