Pyraformer 开源项目使用教程
Pyraformer项目地址:https://gitcode.com/gh_mirrors/py/Pyraformer
1. 项目的目录结构及介绍
Pyraformer 项目的目录结构如下:
Pyraformer/
├── models/
│ ├── LongRange/
│ │ ├── elect/
│ │ │ ├── 168/
│ │ │ │ ├── best_iter0.pth
│ │ │ │ ├── best_iter1.pth
│ │ │ │ ├── best_iter2.pth
│ │ │ │ ├── best_iter3.pth
│ │ │ │ ├── best_iter4.pth
│ │ │ ├── 336/
│ │ │ ├── 720/
│ │ ├── ETTh1/
│ │ ├── ETTm1/
│ ├── SingleStep/
│ │ ├── elect/
│ │ │ ├── best_model.pth
│ │ ├── flow/
│ │ │ ├── best_model.pth
│ │ ├── wind/
│ │ │ ├── best_model.pth
├── scripts/
│ ├── Pyraformer_LR_FC.sh
│ ├── Pyraformer_SS.sh
├── utils/
├── data_loader.py
├── long_range_main.py
├── preprocess_elect.py
├── preprocess_flow.py
├── preprocess_wind.py
├── requirements.txt
├── simulate_sin.py
├── single_step_main.py
├── LICENSE
├── README.md
目录结构介绍
models/
: 存放预训练模型的目录。LongRange/
: 长期预测模型。SingleStep/
: 单步预测模型。
scripts/
: 包含启动脚本。Pyraformer_LR_FC.sh
: 长期预测启动脚本。Pyraformer_SS.sh
: 单步预测启动脚本。
utils/
: 工具函数目录。data_loader.py
: 数据加载模块。long_range_main.py
: 长期预测主文件。preprocess_*.py
: 数据预处理模块。requirements.txt
: 项目依赖文件。simulate_sin.py
: 生成合成数据脚本。single_step_main.py
: 单步预测主文件。LICENSE
: 项目许可证文件。README.md
: 项目说明文档。
2. 项目的启动文件介绍
长期预测启动文件
scripts/Pyraformer_LR_FC.sh
是用于启动长期预测的脚本。使用方法如下:
sh scripts/Pyraformer_LR_FC.sh
单步预测启动文件
scripts/Pyraformer_SS.sh
是用于启动单步预测的脚本。使用方法如下:
sh scripts/Pyraformer_SS.sh
3. 项目的配置文件介绍
依赖文件
requirements.txt
文件列出了项目运行所需的依赖包。使用以下命令安装依赖:
pip install -r requirements.txt
主文件配置
long_range_main.py
: 长期预测主文件,包含命令行参数解析和模型训练逻辑。single_step_main.py
: 单步预测主文件,包含命令行参数解析和模型训练逻辑。
数据预处理配置
preprocess_*.py
文件用于数据预处理,根据不同的数据集进行预处理操作。
以上是 Pyraformer 开源项目的使用教程,涵盖了项目的目录结构、启动文件和配置文件的介绍。希望对您有所帮助!