/GNN

基于pytorch实现的图神经网络训练框架，各个部分模块化，方便修改模型。包含图神经网络模型、训练、验证、测试、wandb可视化。

Primary LanguagePython

pytorch图神经网络预测框架

代码兼容性较强，使用的是一些基本的库、基础的函数
在argparse中可以选择使用wandb，能在wandb网站中生成可视化的训练过程

1，环境

torch：https://pytorch.org/get-started/previous-versions/

pip install tqdm wandb torch_geometric -i https://pypi.tuna.tsinghua.edu.cn/simple

2，数据格式

torch_geometric.data.Data格式组成的列表，并保存为.pt文件，每个元素包含x、edge_index、y、mask等

3，run.py

模型训练时运行该文件，argparse中有对每个参数的说明

4，data_validation.py

用已经训练好的模型对数据进行验证，可以查看哪些数据的损失较大，检查是否有异常

其他

学习笔记：https://github.com/TWK2022/notebook