Pytorch实现GCN(基于PyTorch实现)

前言

大家好,我是阿光。

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

正在更新中~ ✨

Pytorch实现GCN(基于PyTorch实现)_第1张图片

我的项目环境:

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

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


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

一、导入相关库

本项目是采用自己实现的GCN,并没有使用 PyG 库,原因是为了帮助新手朋友们能够对GCN的原理有个更深刻的理解,如果熟悉之后可以尝试使用PyG库直接调用 GCNConv 这个图层即可。

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