GraphTrans 项目使用教程
graphtrans项目地址:https://gitcode.com/gh_mirrors/gr/graphtrans
1. 项目的目录结构及介绍
GraphTrans 项目的目录结构如下:
graphtrans/
├── configs/
│ └── code2/
│ └── gnn-transformer/
│ ├── JK=cat/
│ └── no-virtual/
├── data/
├── models/
├── utils/
├── main.py
├── requirement.yml
├── README.md
└── slurm-run.sh
目录结构介绍
configs/
: 包含项目的配置文件,不同任务和模型的配置文件存放在不同的子目录中。data/
: 用于存放数据集的目录。models/
: 包含项目中使用的各种模型定义。utils/
: 包含项目中使用的各种辅助函数和工具。main.py
: 项目的启动文件。requirement.yml
: 项目的环境依赖文件。README.md
: 项目的说明文档。slurm-run.sh
: 用于在 SLURM 集群上运行项目的脚本。
2. 项目的启动文件介绍
项目的启动文件是 main.py
。该文件负责加载配置、初始化模型、加载数据并启动训练或评估过程。
启动文件的主要功能
- 加载配置文件:根据指定的配置文件路径加载配置。
- 初始化模型:根据配置文件中的参数初始化模型。
- 加载数据:从数据目录中加载训练或评估所需的数据。
- 启动训练或评估:根据配置文件中的参数启动训练或评估过程。
启动示例
# GraphTrans (GCN-Virtual)
python main.py --configs configs/code2/gnn-transformer/JK=cat/pooling=cls+norm_input.yml --runs 5
# GraphTrans (GCN)
python main.py --configs configs/code2/gnn-transformer/no-virtual/pooling=cls+norm_input.yml --runs 5
3. 项目的配置文件介绍
项目的配置文件存放在 configs/
目录下,不同任务和模型的配置文件存放在不同的子目录中。
配置文件示例
# configs/code2/gnn-transformer/JK=cat/pooling=cls+norm_input.yml
dataset: OGBG-Code2
model: GraphTrans
JK: cat
pooling: cls+norm_input
...
配置文件的主要内容
dataset
: 指定使用的数据集。model
: 指定使用的模型。JK
: 指定连接方式。pooling
: 指定池化方法。- 其他参数:包括学习率、批次大小、训练轮数等。
通过配置文件,用户可以灵活地调整模型和训练参数,以适应不同的任务和数据集。