cugraph-pyg API 参考

cugraph-pyg

图存储

cugraph_pyg.data.dask_graph_store.DaskGraphStore(F, ...)	Duck-typed 版本的 PyG 的 GraphStore 和 FeatureStore，使用 Dask 将图结构分布在 GPU 上，并使用 cugraph.gnn.FeatureStore 来存储节点/边特征。
cugraph_pyg.data.graph_store.GraphStore([...])	基于cuGraph的PyG GraphStore实现，将图分布在多个工作节点上。

特征存储

cugraph_pyg.data.feature_store.TensorDictFeatureStore()	一个基本的PyG FeatureStore接口实现，将特征数据存储在单个TensorDict中。
cugraph_pyg.data.feature_store.WholeFeatureStore([...])	一个基本的PyG FeatureStore接口实现，将特征数据存储在WholeGraph WholeMemory中。

数据加载器

cugraph_pyg.loader.dask_node_loader.DaskNeighborLoader(data)	使用Dask通过均匀邻居采样算法采样节点的PyG NeighborLoader接口的Duck-typed版本。
cugraph_pyg.loader.dask_node_loader.BulkSampleLoader(...)	使用Dask和cuGraph执行采样并从磁盘加载采样的小批量的迭代器。
cugraph_pyg.loader.node_loader.NodeLoader(...)	torch_geometric.loader.NodeLoader 的鸭子类型版本。
cugraph_pyg.loader.neighbor_loader.NeighborLoader(...)	Duck-typed 版本的 torch_geometric.loader.NeighborLoader

采样器

cugraph_pyg.sampler.sampler.BaseSampler(...)	方法
cugraph_pyg.sampler.sampler.SampleReader(...)	迭代器，用于处理来自cuGraph分布式采样器的结果。
cugraph_pyg.sampler.sampler.HomogeneousSampleReader(...)	SampleReader的子类，用于读取由cuGraph分布式采样器生成的同构输出样本。
cugraph_pyg.sampler.sampler.SampleIterator(...)	迭代器，将输出图与其特征结合，生成可以输入到GNN模型中的最终输出小批量数据。