一种快速的图形注意力网络。
fastGAT的Python项目详细描述
快速图注意力神经网络
那是什么?在
快速图注意力神经网络(fastGAT)是在GAT(Graph ATtention networks)的基础上发展起来的一种新算法。我们使用angelsh(基于角度的局部敏感度哈希)来设计我们的算法,以获得更快的运行速度和更少的内存使用。 总之,你可以用我们的算法代替GAT,在任何地方都能产生更大的效果。在
如何安装?在
只需使用:
pip install fastGAT
如何使用它?在
快速运行演示。
您只需使用:
^{pr2}$对于简单的demo,它将在CORA数据集中运行我们的方法。在
自定义您自己的超参数。
使用
fromfasGAT.modelimportfastGATasmftblablabla..
自定义您自己的数据集
试试numpy格式矩阵,例如:
adj: the adj matrix
features: feature vector combination of all nodes
labels: label of all nodes
更多信息
有关更多信息,请访问此处:fastGAT
做得好吗?在
你可以欣赏一些图片下面的细节。在
海螺
您可以使用github中的问题发送任何问题和BUG。或者打电话给我:2273067585@qq.com。在
- 项目
标签: