GraphTrans 项目使用教程

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: 指定池化方法。
• 其他参数：包括学习率、批次大小、训练轮数等。

通过配置文件，用户可以灵活地调整模型和训练参数，以适应不同的任务和数据集。

graphtrans项目地址:https://gitcode.com/gh_mirrors/gr/graphtrans