42、图神经网络:GCN、GraphSage与图注意力网络详解

图神经网络:GCN、GraphSage与图注意力网络详解

1. 图卷积网络(GCN)

GCN通过假设λmax = 2简化了之前的表达式,作者期望神经网络参数能够适应这一变化。经过一系列推导,GCN将卷积的输出简化,使得每个GCN滤波器只有一个可学习参数。GCN的滤波操作在拉普拉斯矩阵L上是线性的,它仅从每个节点的一阶邻域收集信息。为了学习包含多跳信息的丰富表示,GCN可以设置多个层。与ChebNet相比,GCN的优势在于它不需要对k - 局部化进行显式参数化,因此不太容易过拟合。在固定的计算预算下,由于其滤波器相对简单,GCN可以设置更深的层。

1.1 使用GCN进行图分类
  • 数据集 :使用MUTAG数据集,该数据集包含188个图,每个图代表一个硝基芳香族化合物,节点表示原子,原子的输入特征是长度为7的独热编码向量“原子类型”,每个图可属于两个类别之一,目标是将预测每个化合物对鼠伤寒沙门氏菌的致突变性作为一个二分类问题。
  • 代码实现
import matplotlib.pyplot as plt
from sklearn.manifold import TSNE
import io, os, sys, types
import pandas as pd
import numpy as np
from stellargraph.mapper import PaddedGraphGenerator
from stellargraph.layer import GCNSupe
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符  | 博主筛选后可见
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值