1 min to read
图神经网络
图卷积操作（GCN）
参考资料
http://tkipf.github.io/graph-convolutional-networks/
https://yzhang1918.github.io/posts/gcn_intuition/
PyTorch Geometric (PyG) 教程
概述
提供了一系列基于图数据的深度学习算法库
数据处理
使用from torch_geometric.data import Data 来描述一张图，Data的关键属性：
- x：节点的特征，维度为：[num_nodes, num_node_features]
- edge_index：边的起止节点，维度为：[2, num_edges]，以稀疏矩阵的方式存储；
- edge_attr：边的属性/特征，维度为：[num_edges, num_edge_features];
- y：标签信息，可能是一张图的类别或者每一个node的类别，维度可以为任意形状；
- pos：节点位置信息，维度：[num_nodes, num_dimensions]
常用方法：
数据前置处理Transformers：
自定义传播逻辑
异构图的网络建模
图中的节点分属不同类型，比如：明星XX出演了《xxx》电视剧；电视剧与明星都是节点，但是属于不同类别的节点；
参考资料
官网：https://pytorch-geometric.readthedocs.io/en/latest/
Comments