机器学习系统与优化 教案 案例16-第四章 GCN用于Cora数据集节点分类案例.docx

案例16——GCN用于Cora数据集节点分类案例

本案例使用图卷积网络（GCN）对Cora数据集中的节点进行分类。

1.数据集

Cora数据集是一个引文网络数据集，包含2733个节点（论文），每个节点有1433个特征（词向量），节点之间有5429条边（引用关系）。数据集分为三个部分：训练集、验证集和测试集，分别包含140、500和1000个节点。

2.代码

importtorch

importtorch.nnasnn

importtorch.nn.functionalasF

fromtorch_geometric.nnimportGCNConv

fromto