在这种背景下,我们提出了一个模块化的、基于Pytorch-Geometric的包Equitorch,希望能够使研究者能够更加灵活地构建等变图神经网络。 对图神经网络相关的部分,如消息传递机制、图数据表示等,我们完全基于Pytorch-Geometric这一广泛使用的图神经网络框架,更方便传统图神经网络的研究者迁移。 径向、角度与球谐)基展开 软截断操作 数学函数equitorch.math 球谐张量、球谐函数、Wigner D矩阵的相关操作 球谐变换与逆球谐变换 数据变换equitorch.transform(基于Pytorch-Geometric
install torch-geometric==2.0.2 -f https://pytorch-geometric.com/whl/torch-1.11.0+cu113.html ---- 参考链接: pytorch-geometric
#179 PyGOD 0.3.0 document Installation PyGOD Github 页面 pytorch-geometric 安装教程 PyTorch与PyTorch Geometric
然而,现有软件实现多局限于单一架构,且主要基于 PyTorch-Geometric、TensorFlow 或 JAX。相比之下,DGL 在大规模图训练中展现出更高的内存效率与速度,适合支撑大批量模拟。
来源:DeepHub IMBA本文约3800字,建议阅读7分钟本文使用pytorch-geometric实现的GNNExplainer作为示例。 本文使用的是pytorch-geometric实现的GNNExplainer作为示例。
这项工作的一个很好的结果是在流行数据集上和在PyTorch-Geometric上的代码基准上的得到了公平的SOTA。
本文使用的是pytorch-geometric实现的GNNExplainer作为示例,有兴趣了解的话可以查看其官方文档 https://pytorch-geometric.readthedocs.io/
这项工作的一个很好的结果是在流行数据集上和在PyTorch-Geometric上的代码基准上的得到了公平的SOTA。
该研究的杰出成果包括:提供了模型在常用数据集上的公平 SOTA 结果和 PyTorch-Geometric 代码,方便以后的论文进行模型对比。
PyTorch-Geometric[20]和DGL[1]用消息传递接口包装了现有的深度学习框架。他们专注于设计一个面向图形的界面来进行改进GNN可编程性,并借鉴了传统图处理系统和DNN框架的优化原理。