文章目录

前言

一、导入相关库

二、加载Cora数据集

三、定义MLP网络

3.1 定义MLP层

3.1.1 定义参数 W W W 和 b b b

3.1.2 定义传播函数

3.1.3 MLP层

3.2 定义MLP网络

四、定义模型

五、模型训练

六、模型验证

七、结果

完整代码

前言

大家好，我是阿光。

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

正在更新中~ ✨

🚨 我的项目环境：

平台：Windows10

语言环境：python3.7

编译器：PyCharm

PyTorch版本：1.11.0

PyG版本：2.1.0

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

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

一、导入相关库

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