PyTorch & PyTorch Geometric图神经网络(GNN)实战

文章预览

【导读】在这篇博文中，我们将使用PyTorch和PyTorch Geometric（PyG），构建图形神经网络框架。 作者| Steeve Huang 除了惊人的速度之外，PyG还提供了一系列精心实现的GNN模型，并在各种论文中进行了说明。 因此，用PyG重现实验将非常方便。 鉴于其在速度和便利性方面的优势，毫无疑问，PyG是最受欢迎和广泛使用的GNN库之一。 让我们深入探讨这个主题吧！ 系统要求 PyTorch - 1.1.0 PyTorch Geometric  -  1.2.0 PyTorch Geometric 基础 本节将介绍PyG的基础知识,包括torch_geometric.data和torch_geometric.nn。 您将学习如何将几何数据传递到GNN，以及如何设计自定义MessagePassing层，GNN的核心。 数据 torch_geometric.data模块包含一个Data类，允许您轻松地从数据中创建图形。 您只需指定： 与每个节点关联的属 ………………………………