动手学图神经网络(9):利用图神经网络进行节点分类 WeightsBiases
利用图神经网络进行节点分类Weights&Biases
引言
在本篇博客中,将深入探讨如何使用图神经网络(GNNs)来完成节点分类任务。以 Cora
数据集为例,该数据集是一个引用网络,节点代表文档,推断每个文档的类别。同时,使用 Weights & Biases(W&B)来跟踪实验过程和结果 。
环境设置
首先, 需要安装必要的包,包括 torch-scatter
、torch-sparse
、pytorch_geometric
和 wandb
。安装完成后,登录到 Weights & Biases。
# Install required packages.
!pip install -q torch