FSNet 项目使用教程

FSNet 项目使用教程

fsnet 项目地址: https://gitcode.com/gh_mirrors/fs/fsnet

1. 项目目录结构及介绍

FSNet 项目的目录结构如下：

fsnet/
├── data/
├── exp/
├── models/
├── scripts/
├── utils/
├── .gitignore
├── CODEOWNERS
├── CODE_OF_CONDUCT.md
├── LICENSE
├── README.md
├── SECURITY.md
├── framework.png
└── main.py

目录介绍

• data/: 存放原始数据文件（CSV 格式）。
• exp/: 存放实验结果和日志文件。
• models/: 存放模型定义和实现文件。
• scripts/: 存放运行实验的脚本文件。
• utils/: 存放工具函数和辅助代码。
• .gitignore: Git 忽略文件配置。
• CODEOWNERS: 代码所有者配置文件。
• CODE_OF_CONDUCT.md: 行为准则文件。
• LICENSE: 项目许可证文件。
• README.md: 项目介绍和使用说明文件。
• SECURITY.md: 安全策略文件。
• framework.png: 项目框架图。
• main.py: 项目主启动文件。

2. 项目启动文件介绍

main.py

main.py 是 FSNet 项目的主启动文件，负责初始化模型、加载数据、训练和评估模型。以下是该文件的主要功能模块：

• 数据加载: 从 data/ 目录加载原始数据。
• 模型初始化: 根据配置文件初始化 FSNet 模型。
• 训练: 执行模型的训练过程，支持多种训练策略（如 OGD、ER、DERPP 等）。
• 评估: 在测试集上评估模型的性能。

使用方法

python main.py --method fsnet --data ETT --test_bsz 1 --seq_len 60 --pred_len 1

3. 项目的配置文件介绍

FSNet 项目没有显式的配置文件，但可以通过命令行参数进行配置。以下是一些常用的配置参数：

• --method: 指定训练策略，可选值包括 ogd, large, er, derpp, nomem, naive, fsnet。
• --data: 指定数据集，可选值包括 ETT, ECL, Traffic, WTH。
• --test_bsz: 测试时的批量大小，必须设置为 1 以支持在线学习。
• --seq_len: 回溯窗口的长度，默认设置为 60。
• --pred_len: 预测窗口的长度，默认设置为 1。

示例配置

python main.py --method fsnet --data ETT --test_bsz 1 --seq_len 60 --pred_len 1

以上命令将使用 FSNet 方法在 ETT 数据集上进行在线时间序列预测，测试批量大小为 1，回溯窗口长度为 60，预测窗口长度为 1。

fsnet 项目地址: https://gitcode.com/gh_mirrors/fs/fsnet