cugraph_pyg.sampler.BaseSampler

class cugraph_pyg.sampler.BaseSampler(

sampler: DistributedNeighborSampler,

data: Tuple[FeatureStore, GraphStore],

batch_size: int = 16,

)

Methods

sample_from_edges
sample_from_nodes

__init__(

sampler: DistributedNeighborSampler,

data: Tuple[FeatureStore, GraphStore],

batch_size: int = 16,

)

Methods

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