cugraph_pyg.sampler.sampler.HomogeneousSampleReader#

class cugraph_pyg.sampler.sampler.HomogeneousSampleReader(base_reader: DistSampleReader)[source]#

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

__init__(base_reader: DistSampleReader)[source]#

Constructs a new HomogeneousSampleReader

Parameters:
base_reader: DistSampleReader

The reader responsible for loading saved samples produced by the cuGraph distributed sampler.

Methods

__init__(base_reader)

Constructs a new HomogeneousSampleReader