dgl使用ogb包导入OGB数据集

静静喜欢大白

已于 2022-04-07 11:49:24 修改

阅读量2.7k

点赞数

分类专栏： dgl 文章标签： dgl

于 2020-12-29 12:43:38 首次发布

本文链接：https://blog.csdn.net/lj2048/article/details/111893660

版权

dgl 专栏收录该内容

30 篇文章 ¥49.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何使用DGL结合ogb包来导入Open Graph Benchmark (OGB)的数据集，涵盖了Graph Property Prediction、Node Property Prediction和Link Property Prediction任务的数据加载方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

OGB简介

Open Graph Benchmark (OGB) 是一个图深度学习的基准数据集。官方的 ogb 包提供了用于下载和处理OGB数据集到 dgl.data.DGLGraph 对象的API。本节会介绍它们的基本用法。

使用pip安装ogb包

pip install ogb

**为 Graph Property Prediction 任务加载数据集**

# 载入OGB的Graph Property Prediction数据集
import dgl
import torch
from ogb.graphproppred import DglGraphPropPredDataset
from torch.utils.data import DataLoader

def _collate_fn(batch):
    # 小批次是一个元组(graph, label)列表
    graphs = [e[0] for e in batch]
    g = dgl.batch(graphs)
    labels = [e[1] for e in batch]
    label

了解本专栏