ADGAT 项目使用教程

1. 项目的目录结构及介绍

ADGAT 项目的目录结构如下：

ADGAT/
├── data/
│   └── raw_data/
│       ├── stocks.txt
│       └── relations/
├── Layers.py
├── Model.py
├── README.md
├── main.py
├── utils.py
└── LICENSE

目录结构介绍：

• data/: 存放预处理后的数据文件。
  • raw_data/: 存放原始数据文件。
    • stocks.txt: 包含选定的198个股票代码。
    • relations/: 存放公司关系数据。
• Layers.py: 包含模型中的层定义。
• Model.py: 包含模型的主要实现。
• README.md: 项目的说明文档。
• main.py: 项目的启动文件。
• utils.py: 包含一些辅助函数。
• LICENSE: 项目的许可证文件。

2. 项目的启动文件介绍

main.py

main.py 是 ADGAT 项目的启动文件。它负责加载数据、初始化模型、进行训练和评估。

主要功能：

• 数据加载: 从 data/ 目录中加载预处理后的数据。
• 模型初始化: 使用 Model.py 中定义的模型进行初始化。
• 训练和评估: 执行模型的训练和评估过程。

使用方法：

$ python main.py --device=0

• --device=0: 指定使用 GPU 进行计算。

3. 项目的配置文件介绍

ADGAT 项目没有明确的配置文件，但可以通过命令行参数在 main.py 中进行配置。

主要配置项：

• --device: 指定使用的设备（如 GPU 或 CPU）。
• --data_path: 指定数据文件的路径（默认使用 data/ 目录中的数据）。

示例：

$ python main.py --device=0 --data_path=./data/

通过这种方式，用户可以根据需要调整项目的配置。