Samplers#

cugraph_pyg.sampler.BaseSampler(sampler, data)

cugraph_pyg.sampler.SampleIterator(data, ...)

Iterator that combines output graphs with their features to produce final output minibatches that can be fed into a GNN model.

cugraph_pyg.sampler.BaseDistributedSampler(...)

Base class for distributed graph sampling using cuGraph.

cugraph_pyg.sampler.DistributedNeighborSampler(...)

cugraph_pyg.sampler.sampler.SampleReader(...)

Iterator that processes results from the cuGraph distributed sampler.

cugraph_pyg.sampler.sampler.HomogeneousSampleReader(...)

Subclass of SampleReader that reads homogeneous output samples produced by the cuGraph distributed sampler.

cugraph_pyg.sampler.sampler.HeterogeneousSampleReader(...)

Subclass of SampleReader that reads heterogeneous output samples produced by the cuGraph distributed sampler.

cugraph_pyg.sampler.io.BufferedSampleReader(...)