PyTorch Geometric (PyG)是构建图神经网络模型和实验各种图卷积的主要工具。在本文中我们将通过链接预测来对其进行介绍。
链接预测答了一个问题:哪两个节点应该相互链接?我们将通过执行“转换分割”，为建模准备数据。为批处理准备专用的图数据加载器。在Torch Geometric中构建一个模型，使用PyTorch Lightning进行训练，并检查模型的性能。
https://avoid.overfit.cn/post/e14c4369776243d68c22c4a2a0346db2