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