
基于PyTorch的MTS-Mixers代码实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目基于PyTorch框架实现了MTS-Mixers算法,为多时间序列数据处理提供高效解决方案,适用于时间序列预测与分析任务。
MTS-Mixers是通过因子化时间和通道混合进行多元时间序列预测的官方实现方法。使用步骤如下:
1. 安装Python版本需不低于3.6以及PyTorch版本不低于1.5.0。
2. 运行命令`pip install -r requirements.txt`安装依赖库。
3. 将数据下载后放置在项目的`./dataset`文件夹中,所有基准测试的数据集都经过了良好的预处理,并且可以轻松使用。这些数据可以从Google Drive获取(此处仅说明来源)。
4. 训练模型时,请参考script.md中的脚本示例运行所有的基准测试;如果需要调整超参数,则可以在run.py中查看更多关于配置的信息。
引用文献:Li Z, Rao Z, Pan L, et al. MTS-Mixers: Multivariate Time Series Forecasting via Factorized Temporal and Channel Mixing[J]. arXiv preprint arXiv:2302.04501, 2023.
全部评论 (0)
还没有任何评论哟~


