dgl.dataloading
dgl.dataloading 包提供了两个基本原语,用于构建从图数据加载的数据管道。Sampler
代表从原始图生成子图样本的算法,而 DataLoader
代表对这些样本进行迭代的迭代器。
DGL 提供了一些内置的采样器,它们是 Sampler
的子类。创建新的采样器遵循相同的范例。阅读我们的用户指南章节 第6章:大规模图上的随机训练 以获取更多示例和解释。
整个包仅适用于 PyTorch 后端。
DataLoaders
采样图数据加载器。 |
|
批处理图数据加载器。 |
Samplers
图采样器的基类。 |
|
通过邻居采样为多层 GNN 构建节点表示计算依赖关系的采样器。 |
|
通过 Labor 采样为多层 GNN 构建节点表示计算依赖关系的采样器,来自 NeurIPS 2023 论文 Layer-Neighbor Sampling -- Defusing Neighborhood Explosion in GNNs。 |
|
通过从所有邻居获取消息为多层 GNN 构建节点表示计算依赖关系的采样器。 |
|
来自 Cluster-GCN: 一种训练深度大型图卷积网络的有效算法 的聚类采样器。 |
|
来自 Deep Graph Neural Networks with Shallow Subgraph Samplers 的 K 跳子图采样器。 |
|
来自 GraphSAINT: 基于图采样的归纳学习方法 的随机节点/边/游走采样器。 |
Sampler 转换
|
从节点级采样器创建边级采样器。 |
|
用于以消息传递流图 (MFGs) 形式采样 mini-batch 的基类。 |
链路预测的负采样器
|
|
根据均匀分布为每个源节点随机选择负目标节点的负采样器。 |
|
根据均匀分布随机选择负源-目标对的负采样器。 |
特征预取的实用类和函数
为输入图的 |
|
为输入图的 |
|
为输入图的 |
|
为输入图的 |
|
特征预取的占位符。 |