This is the official implementation of paper "MSTVI: Multi-Scale Time-Variable Interaction for Multivariate Time Series Forecasting".
- Install Python 3.10. For convenience, execute the following command.
pip install -r requirements.txt
-
Prepare Data. You can obtain the well pre-processed datasets from [Google Drive]. Then place the downloaded data in the folder
./dataset
. -
Train and evaluate model. We provide the experiment scripts for all benchmarks under the folder
./scripts/
. You can reproduce the experiment results as the following examples:
# long-term forecast
bash scripts/long_term_forecast/MSTVI/traffic.sh
If you have any questions or suggestions, feel free to contact our maintenance team:
- Quangao Liu ([email protected])
- Ruiqi Li ([email protected])
- Maowei Jiang ([email protected])
Or describe it in Issues.
If you find this repo useful, please cite our paper
@inproceedings{liu2024mstvi,
title={MSTVI: Multi-Scale Time-Variable Interaction for Multivariate Time Series Forecasting},
author={Quangao Liu and Ruiqi Li and Maowei Jiang and Wei Yang and Cheng Liang and Zhuozhang Zou},
year={2024},
}
Our code is based on Time Series Library (TSLib):https://github.com/thuml/Time-Series-Library