Graph 注意力网络 v2 (GATv2)

这是 Gatv2 运营商的 PyTorch 实现,来自论文 G raph 注意力网络有多专心?

GATV2 处理图形数据。图由连接节点的节点和边组成。例如,在 Cora 数据集中,节点是研究论文,边缘是连接论文的引文。

G@@

ATv2 运算符修复了标准 GAT 的静态注意力问题:由于标准 GAT 中的线性层紧随其后应用,因此参与节点的排名在查询节点上是无条件的。相比之下,在 GATv2 中,每个节点都可以参与任何其他节点。

以下是在 Cora 数据集上训练双层 GATv2 的训练代码

View Run