STGCN项目安装与使用教程

STGCN项目安装与使用教程

STGCN The PyTorch implementation of STGCN. 项目地址: https://gitcode.com/gh_mirrors/stg/STGCN

1. 项目目录结构及介绍

STGCN项目的目录结构如下：

stgcn/
├── data/                # 存放数据集
├── figure/              # 存放图表
├── model/               # 模型相关代码
├── script/              # 脚本文件，包括训练和测试等
├── .gitignore           # git忽略文件
├── LICENSE              # 项目许可证
├── README.md            # 项目说明文件
├── main.py              # 项目主程序
├── requirements.txt     # 项目依赖

• data/：该目录用于存放项目所需的数据集，例如METR-LA和PEMS-BAY等交通数据。
• figure/：存放项目生成的图表，如模型性能曲线等。
• model/：包含STGCN模型的结构代码，包括图卷积层、时序卷积层等。
• script/：包含各种脚本文件，例如数据预处理脚本、模型训练脚本、模型评估脚本等。
• .gitignore：指定git版本控制时需要忽略的文件和目录。
• LICENSE：项目使用的许可证信息，本项目采用LGPL-2.1许可证。
• README.md：项目的说明文档，介绍项目的基本信息和如何使用。
• main.py：项目的主程序文件，用于启动模型训练或测试。
• requirements.txt：列出项目运行所需的Python库依赖。

2. 项目的启动文件介绍

项目的启动文件是main.py。该文件负责初始化模型、加载数据、设置训练参数以及启动训练或测试过程。以下是一个简化的main.py文件结构：

import argparse
from model.stgcn import STGCN
from script.train import train
from script.test import test

def main():
    parser = argparse.ArgumentParser()
    # 添加命令行参数
    parser.add_argument('--mode', type=str, default='train')
    # 解析命令行参数
    args = parser.parse_args()

    # 根据模式选择训练或测试
    if args.mode == 'train':
        model = STGCN()
        train(model)
    elif args.mode == 'test':
        model = STGCN()
        test(model)

if __name__ == "__main__":
    main()

用户可以通过命令行传递参数来指定程序是进行训练还是测试。

3. 项目的配置文件介绍

项目的配置文件通常存放在script/目录下，如config.py。配置文件用于定义项目运行时的参数，如数据集路径、模型参数、训练参数等。以下是一个示例配置文件的结构：

# 配置文件示例 config.py

# 数据集相关配置
data_config = {
    'data_path': 'data/METR-LA/',
    'train_ratio': 0.8,
    'val_ratio': 0.1,
    'test_ratio': 0.1
}

# 模型相关配置
model_config = {
    'in_channels': 1,
    'out_channels': 1,
    'num_stations': 207,
    'embedding_dim': 32,
    'dropout': 0.3
}

# 训练相关配置
train_config = {
    'batch_size': 64,
    'learning_rate': 0.001,
    'epochs': 100,
    'early_stop': 10
}

# 测试相关配置
test_config = {
    'batch_size': 64
}

在项目的代码中，可以通过import config来访问这些配置参数，从而方便地调整项目设置。

STGCN The PyTorch implementation of STGCN. 项目地址: https://gitcode.com/gh_mirrors/stg/STGCN