Pytorch实现GCN(基于Message Passing消息传递机制实现)

文章目录

• 前言
• 一、导入相关库
• 二、加载Cora数据集
• 三、定义GCN网络
• • 3.1 定义GCN层
  • • 3.1.1 消息传递阶段（message）
    • 3.1.2 消息聚合阶段（aggregate）
    • 3.1.3 节点更新阶段（update）
    • 3.1.4 定义传播过程（propagate）
    • 3.1.5 定义GCNConv层
  • 3.2 定义GCN网络
• 四、定义模型
• 五、模型训练
• 六、模型验证