home graphs gat
View code on Github
这是 PyTorch 对《图形注意力网络》论文的实现。
GAT 处理图表数据。图由节点和连接节点的边组成。例如,在 Cora 数据集中,节点是研究论文,边缘是连接论文的引文。
GAT 使用蒙面自注意力,有点类似于变形金刚。GAT 由相互堆叠的图表注意力层组成。每个图注意力层都将节点嵌入作为转换后的嵌入的输入和输出获得节点。节点嵌入会注意它所连接的其他节点的嵌入。图形注意力层的详细信息与实现一起包括在内。
以下是在 Cora 数据集上训练两层 GAT 的训练代码。