MrCAN

This code is the implementation of the paper "MrCAN: Multi-relations aware convolutional attention network for multivariate time series forecasting".

Requirements

The recommended requirements for MrCAN are specified as follows:

• Python 3.8
• torch==1.13.1
• numpy==1.22.4
• pandas==1.5.3
• scikit-learn==1.2.1

The dependencies can be installed by:

pip install -r requirements.txt

Data

The datasets can be obtained and put into dataset/ folder

• SML2010
• Highways Traffic
• PM2.5
• EEG
• ETT

Usage

To train and evaluate MrCAN on a dataset, run the following command:

python main.py