Graph U-Nets 项目使用教程

1. 项目目录结构及介绍

gunet/
├── configs/
│   ├── config.yaml
│   └── ...
├── data/
│   ├── README.md
│   └── ...
├── doc/
│   └── ...
├── src/
│   ├── utils/
│   │   ├── ops.py
│   │   └── ...
│   └── ...
├── .gitignore
├── LICENSE
├── README.md
└── run_GNN.sh

目录结构介绍

• configs/: 存放项目的配置文件，如 config.yaml。
• data/: 存放数据集文件，包含数据集的格式说明文件 README.md。
• doc/: 存放项目的文档文件。
• src/: 项目的源代码目录，包含主要的实现文件，如 utils/ops.py。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目的开源许可证文件。
• README.md: 项目的说明文档。
• run_GNN.sh: 项目的启动脚本。

2. 项目启动文件介绍

run_GNN.sh

run_GNN.sh 是项目的启动脚本，用于运行 Graph U-Nets 模型。脚本的使用方法如下：

./run_GNN.sh DATA FOLD GPU

• DATA: 指定要使用的数据集，如 DD。
• FOLD: 指定交叉验证的折数，范围为 1-10。
• GPU: 指定要使用的 GPU 编号。

例如，要在 DD 数据集上使用 10 折交叉验证并在 GPU #0 上运行，可以使用以下命令：

./run_GNN.sh DD 0 0

3. 项目的配置文件介绍

configs/config.yaml

config.yaml 是项目的主要配置文件，用于配置模型的参数和运行环境。配置文件的内容可能包括以下部分：

# 数据集配置
dataset:
  name: DD
  path: data/DD

# 模型配置
model:
  type: GraphUNet
  hidden_dim: 64
  num_layers: 5

# 训练配置
training:
  epochs: 100
  batch_size: 32
  learning_rate: 0.001

# GPU 配置
gpu:
  device: 0

配置文件介绍

• dataset: 配置数据集的相关参数，如数据集名称和路径。
• model: 配置模型的相关参数，如模型类型、隐藏层维度、层数等。
• training: 配置训练过程的相关参数，如训练轮数、批量大小、学习率等。
• gpu: 配置 GPU 设备的相关参数。

通过修改 config.yaml 文件，可以调整模型的训练和运行参数，以适应不同的需求和环境。