Pytorch-geometric 图神经网络库 教程 (一)
前言
之前我是一直使用DGL这个图神经网络库做图神经网络方面的研究的,DGL也是一个特别漂亮的图神经网络开源库。但是DGL有如下两个缺点:
DGL适合跑那种邻接矩阵是良性定义的graph。良性定义是指Aij∈{0,1}A_{ij}\in\{0,1\}Aij∈{0,1},邻居矩阵的每个元素不能是连续的。
DGL的库更新的慢。
此外,DGL主要是基于传统的图计算那套方法来实现的,里面有mailbox之类的,并不是用数学公式来直接表达。
Pytorch-geometric则是可以解决如上的两个缺点。
因此,
原创
2021-04-05 21:01:00 ·
3281 阅读 ·
1 评论