グラフ・アテンション・ネットワークス v2 (GATv2)

これは、「グラフアテンションネットワークはどの程度注意深いのか?」という論文のGATv2演算子をPyTorchで実装したものです

GATv2はグラフデータを処理します。グラフは、ノードとノードを接続するエッジで構成されます。たとえば、Coraデータセットでは、ノードは研究論文で、端は論文をつなぐ引用です

GATv2 演算子は、標準 GAT の静的注意の問題を解決します。標準 GAT の線形レイヤーは次々に適用されるため、参加ノードのランク付けはクエリノードで条件付けされません。対照的に、GATv2では、すべてのノードが他のノードに接続できます

これは、Coraデータセットで2層GATv2をトレーニングするためのトレーニングコードです