发布时间:2023-04-19 文章分类:电脑基础 投稿人:樱花 字号: 默认 | | 超大 打印

PyTorch+PyG实现图神经网络经典模型目录

前言

大家好,我是阿光。

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

正在更新中~ ✨

PyTorch+PyG实现图神经网络经典模型目录

🚨 我的项目环境:

🌠 『精品学习专栏导航帖』

对于本专栏的网络模型,分别使用了三种实现方式 PyG框架实现PyTorch实现Message Passing消息传递机制实现,小伙伴可以按照自己的能力以及需求学习不同的实现方式。

注意 🚨:本目录中已存在的链接博文已全部写好,例如 + (一):节点分类 这类带有删除线的文章表示正在更新中,如果写完会去掉删除线,点击出现404表示文章还没有发布,后续根据情况陆续发布。

🌈『目录』

📢 PyG算子、数据集介绍

📢 图神经网络常见任务与应用场景

📢 图嵌入学习(Graph Embedding)

📢 图池化(Graph Pooling)

📢 MLP

📢 GCN

📢 GAT

📢 GIN

📢 GraphSAGE

📢 EdgeCNN

📢 GraphConv

注意🚨:所有文章使用的图数据是经典的 Cora 数据集,定义的训练轮数(200轮)以及损失函数优化器都是一致的,由于图网络很容易过拟合导致训练集的分类精度达到 99.9%,所以下表中显示的数据都是基于测试集的。

Accuracy Loss
MLP 0.1800 1.9587
GCN 0.7200 1.3561
GAT 0.7810 1.0362
GIN 0.7650 0.9645
GraphSAGE 0.7060 1.2712
EdgeCNN 0.3790 1.7529
GraphConv 0.6030 1.2378