Skip to content

Repository files navigation

BiTGraph (ICLR 2024)

The code for paper: Biased Temporal Convolution Graph Network for Time Series Forecasting with Missing Values.

Getting Start

  1. Install requirements. pip install -r requirements.txt
  2. Download data.

Download Metr-LA, ETTh1, Electricity, PEMS datasets from here. Obtain BeijingAir dataset from Brits. Put all the files in the ./data.

  1. Training.

python main.py --epochs 200 --mask_ratio 0.2 --dataset-name Metr

  1. Testing.

python test_forecasting.py --epochs 200 --mask_ratio 0.2 --dataset-name Metr

Citation

@inproceedings{BiTGraph, title={Biased Temporal Convolution Graph Network for Time Series Forecasting with Missing Values},
author={Chen, Xiaodan and Li, Xiucheng and Liu, Bo and Li, Zhijun},
booktitle={International Conference on Learning Representations (ICLR)},
year={2024}
}

About

The code for Biased Temporal Convolution Graph Network for Time Series Forecasting with Missing Values.

Resources

Stars

34 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages