Spectral Temporal Graph Neural Network for Multivariate Time-series Forecasting
StemGNN
This repository is the official implementation of Spectral Temporal Graph Neural Network for Multivariate Time-series Forecasting.
Requirements
Recommended version of OS & Python:
To install python dependencies, virtualenv is recommended, sudo apt install python3.7-venv to install virtualenv for python3.7. All the python dependencies are verified for pip==20.1.1 and setuptools==41.2.0. Run the following commands to create a venv and install python dependencies:
python3.7 -m venv venv
source venv/bin/activate
pip install --upgrade pip
pip install -r requirements.txt
Datasets
PEMS03,
PEMS04,
PEMS07,
PEMS08,
METR-LA,
PEMS-BAY,
Solar,
Electricity,
ECG5000,
COVID-19
We can get the raw data through the links above. We evaluate the performance of traffic flow forecasting on PEMS03, PEMS07, PEMS08 and traffic speed forecasting on PEMS04, PEMS-BAY and METR-LA. So we use the traffic flow