GAT学习：PyG实现GAT（使用PyG封装好的GATConv函数）（三）

最新推荐文章于 2025-06-04 20:58:26 发布

原创

最新推荐文章于 2025-06-04 20:58:26 发布 · 1.2w 阅读

47 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文介绍了如何利用PyG库中的GATConv模块，以简化和提升图注意力网络（GAT）的实现效果。在Cora数据集上的实验表明，使用封装的GATConv能获得约65%的准确率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

之前我的实现方式相对而言麻烦且准确率不够好，只能达到65%左右的准确率（Cora上），这里介绍直接用PyG封装好的GAT函数实现：

import torch
import math
from torch_geometric.nn import MessagePassing
from torch_geometric.nn import GATConv

from torch_geometric.utils import add_self_loops,degree
from torch_geometric.datasets import Planetoid
import ssl
import torch.nn.functional as F


class Net(torch.nn.Module):
    def __init__(self):
        super(Net,self).__init__()
        self.gat1=GATConv(dataset.num_node_features,