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

前言

大家好,我是阿光。

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

正在更新中~ ✨

Pytorch实现GCN(基于Message Passing消息传递机制实现)_第1张图片

我的项目环境:

  • 平台:Windows10
  • 语言环境:python3.7
  • 编译器:PyCharm
  • PyTorch版本:1.11.0
  • PyG版本:2.1.0

项目专栏:【图神经网络代码实战目录】


本文我们将使用PyTorch来简易实现一个GCN(图卷积网络),不使用PyG库,让新手可以理解如何PyTorch来搭建一个简易的图网络实例demo。

一、导入相关库

本项目是采用自己实现的GCN,使用的是 图神经网络生成节点表征的范式–消息传递(Message Passing)范式,它将卷积算子推广到了不规则数据领域,实现了图与神经网络的连接,消息传递范式由于简单、强大的特性,现已被人们广泛地使用。

 
 

你可能感兴趣的:(图神经网络代码实战,pytorch,深度学习,python,人工智能,神经网络)