cugraph-pyg API 参考

cugraph-pyg

图存储

cugraph_pyg.data.dask_graph_store.DaskGraphStore(F, ...)	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)	PyG NeighborLoader 接口的鸭子类型实现，使用 Dask 通过统一邻居采样算法对节点进行采样。
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(...)	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 模型）的迭代器。