基于PyG库的多种类图神经网络对比研究

作者:周波; 周微; 毛勇超
来源:深圳职业技术学院学报, 2022, 21(03): 21-49.
DOI:10.13899/j.cnki.szptxb.2022.03.003

摘要

各种各样的基于消息传递机制的图神经网络架构被提出,但是这些图神经网络在同一节点分类任务的表现性能如何缺少相应的研究.文章基于PyG图神经网络库,编写实验代码,包括数据集处理、图神经网络搭建和节点分类模型训练3个主要环节,实现对不同架构的图神经网络在节点分类任务上的对比.通过利用多指标衡量基于多种类的图神经网络构建的节点分类模型,最后得出最适合节点分类的图神经网络结构,并给出实验分析,同时提出了今后的研究方向.

全文