Pytorch实现EdgeCNN(基于PyTorch实现)

文章目录

• 前言
• 一、导入相关库
• 二、加载Cora数据集
• 三、定义EdgeCNN网络
• • 3.1 定义EdgeConv层
  • • 3.1.1 特征拼接
    • 3.1.2 max聚合
    • 3.1.3 特征映射
    • 3.1.4 EdgeConv层
  • 3.2 定义EdgeCNN网络
• 四、定义模型
• 五、模型训练
• 六、模型验证
• 七、结果
• 完整代码

前言

大家好，我是阿光。

本专栏整理了《图神经网络代码实战》，内包含了不同图神经网络的相关代码实现（PyG以及自实现），理论与实践相结合，如GCN、GAT、GraphSAGE等经典图网络，每一个代码实例都附带有完整的代码。

正在更新中~ ✨