#542 · 主分类: 教育与研究

ogb

datasets deep-learning graph-machine-learning graph-neural-networks

用于图机器学习的基准数据集、数据加载器和评估器

项目最后更新:05/06/25

GitHub Stars

2.1K

Forks数量

406

贡献者数量

32

许可证

MIT

收录理由

做图机器学习研究时,最头疼的往往不是模型本身,而是找到一套靠谱的数据集和评测口径。OGB 把几十个标准化的图数据集收拢到一起,覆盖节点、连边、整图三类预测任务,来源包括分子性质预测、社交与信息网络、异质知识图谱等真实场景。它的数据加载器直接对接 PyTorch Geometric 或 DGL,几行代码就能完成下载,并自动套用官方划分的训练、验证、测试集;每个数据集还自带评估器,返回统一的指标。这样一来,不同论文、不同项目之间的结果就有了可比性。无论你是要写图机器学习论文,还是正在搭 GNN 原型,从灵感到一个可信的基线数字,OGB 都是眼下最快的路径。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目