生物医学网络科学Biomedical Network Science

PyTorch Geometric(PyG)

基于 PyTorch 的图神经网络库。

访问官方网站 ↗

领域:计算方法与研究规范 · 分子网络与基因调控 相关专题:结果呈现与研究前沿、疾病模块、网络邻近度与药物重定位

适合的研究问题

在分子图、PPI 网络或脑连接组上训练图神经网络,完成节点分类、链接预测与图分类。

数据类型

Python 库,实现多种图卷积与消息传递层,内置大量基准图数据集。

获取方式

pip install torch_geometric(需先安装 PyTorch)

访问限制与许可

MIT。

相关工具与格式

PyTorch、NetworkX、RDKit

引用要求

Fey M, Lenssen JE. Fast graph representation learning with PyTorch Geometric. ICLR 2019 Workshop(arXiv:1903.02428)。