OpenTS-Bench 快速开始
OpenTS-Bench 为您提供三大关键功能:
- 评测您的算法:根据标准化数据集评估您自己新开发的算法/模型。
- 测试自定义数据集:验证现有算法在您的自定义数据集上的表现。
- 跟踪算法排名:在我们的交互式排行榜上监控不同设置下的性能比较。
目录
快速开始
下载
基于 PyPI
创建一个Python环境 (注意: 该项目基于python 3.8评测), 使用以下命令下载依赖:
1 | pip install -r requirements.txt |
基于 Docker
我们提供 Dockerfile 来配置环境. 为此首先需要安装Docker服务.
1 | docker build . -t tfb:latest |
1 | docker run -it -v $(pwd)/:/app/ tfb:latest bash |
数据准备
下载是数据放置在./dataset
文件夹.
训练与评测
所有的实验脚本在文件夹 ./scripts/multivariate_forecast
, 与 ./scripts/univariate_forecast
之中. 例如可以通过以下命令来复现结果:
1 | sh ./scripts/multivariate_forecast/ILI_script/DLinear.sh |
评测您的算法
我们提供了自定义方法的教程: tutorial.
测试自定义数据集
我们提供了自定义数据集的教程: tutorial.
常见问题
如何在Pycharm中运行代码?
在pycharm下运行时,请将双引号转义并删除空格与首尾的单引号。
例如: ‘{“d_ff”: 512, “d_model”: 256}’ —> {\“d_ff\“:512,\“d_model\“:256}
1 | --config-path "rolling_forecast_config.json" --data-name-list "ILI.csv" --strategy-args {\"horizon\":24} --model-name "time_series_library.DLinear" --model-hyper-params {\"batch_size\":16,\"d_ff\":512,\"d_model\":256,\"lr\":0.01,\"horizon\":24,\"seq_len\":104} --adapter "transformer_adapter" --gpus 0 --num-workers 1 --timeout 60000 --save-path "ILI/DLinear" |
引用
如果您发现我们的工作有帮助,欢迎引用我们的论文。
1 | @inproceedings{qiu2024tfb, |
1 | @inproceedings{qiu2025tab, |
1 | @inproceedings{li2024TSMF-Bench, |
联系我们
若有任何问题,欢迎联系
- Xiangfei Qiu (xfqiu@stu.ecnu.edu.cn)