cugraph_pyg.sampler.sampler.BaseSampler#

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

Methods

sample_from_edges

sample_from_nodes

__init__(sampler: DistSampler, 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)