cugraph-pyg API Reference

cugraph-pyg

Graph Storage

cugraph_pyg.data.dask_graph_store.DaskGraphStore(F, ...)	Duck-typed version of PyG's GraphStore and FeatureStore.
cugraph_pyg.data.graph_store.GraphStore([...])	This object uses lazy graph creation.

Feature Storage

cugraph_pyg.data.feature_store.TensorDictFeatureStore()	A basic implementation of the PyG FeatureStore interface that stores feature data in a single TensorDict.
cugraph_pyg.data.feature_store.WholeFeatureStore([...])	A basic implementation of the PyG FeatureStore interface that stores feature data in WholeGraph WholeMemory.

Data Loaders

cugraph_pyg.loader.dask_node_loader.DaskNeighborLoader(data)	Attributes:
cugraph_pyg.loader.dask_node_loader.BulkSampleLoader(...)
cugraph_pyg.loader.node_loader.NodeLoader(...)	Duck-typed version of torch_geometric.loader.NodeLoader
cugraph_pyg.loader.neighbor_loader.NeighborLoader(...)	Node loader that implements the neighbor sampling algorithm used in GraphSAGE.

Samplers

cugraph_pyg.sampler.sampler.BaseSampler(...)	Methods
cugraph_pyg.sampler.sampler.SampleReader(...)
cugraph_pyg.sampler.sampler.HomogeneousSampleReader(...)
cugraph_pyg.sampler.sampler.SampleIterator(...)