文件名称:PyTorch图机器学习Benchmark集-python
文件大小:4.86MB
文件格式:ZIP
更新时间:2024-07-08 12:59:12
机器学习
PyTorch图机器学习Benchmark集 概述 开放图基准 (OGB) 是用于图机器学习的基准数据集、数据加载器和评估器的集合。 数据集涵盖了各种图形机器学习任务和实际应用。 OGB 数据加载器与流行的图形深度学习框架完全兼容,包括 PyTorch Geometric 和 Deep Graph Library (DGL)。 它们提供自动数据集下载、标准化的数据集拆分和统一的性能评估。 OGB 旨在提供涵盖重要图机器学习任务、多样化数据集规模和丰富领域的图数据集。 图 ML 任务:我们涵盖了三个基本的图机器学习任务:节点、链接和图级别的预测。 规模多样化:小型图数据集可以在单个 GPU 内处理,而中大型图可能需要多个 GPU 或巧妙的采样/分区技术。 丰富的领域:图数据集来自不同的领域,从科学领域到社会/信息网络,还包括异构知识图。 OGB 是一项持续的努力,我们计划在未来扩大我们的覆盖范围。 安装 您可以使用 Python 的包管理器 pip 安装 OGB。 如果您之前安装过 ogb,请确保将版本更新到 1.2.0。 发行说明可在此处获得。 要求 Pytho