动手学PyG(五)：第一个GNN

在前几部分的学习后，我们已经了解了PyG上图的表示以及内置数据库的使用。接下来可以创建第一个图神经网络了！

这里将会从一个GCN层开始，并在Cora数据集上测试其功能。对于GCN的详细介绍参见这里

首先导入Cora数据集

from torch_geometric.datasets import Planetoid

dataset = Planetoid(root='/tmp/Cora', name='Cora')
>>> Cora()

注意这里并没有分batch。接下来让我们创建一个两层的GCN。

import torch
import torch.nn.functional as F
from torch_geometric.nn ipmort GCNConv

class GCN(torch.nn.Module):
	def __init__(self):
		super()