pyg-team
★ 23,960pytorch_geometric
PyTorch 的图神经网络库。为此,我们加载 Cora 数据集,并使用预定义的 GCNConv 创建一个简单的 2 层 GCN 模型:我们现在可以在训练循环中优化模型,类似于标准 PyTorch 训练过程。
HYSEN LABS 项目目录
来自 pyg-team 组织的可信仓库、分类与深度解析。
PyTorch 的图神经网络库。为此,我们加载 Cora 数据集,并使用预定义的 GCNConv 创建一个简单的 2 层 GCN 模型:我们现在可以在训练循环中优化模型,类似于标准 PyTorch 训练过程。