DASTNet

CIKM 2022: Domain Adversarial Spatial-Temporal Network: A Transferable Framework for Short-term Traffic Forecasting across Cities

Requirements

• python == 3.6
• torch == 1.7.0+cu110
• networkx == 2.4
• gensim == 3.8.1

See requirements.txt for more details.

Dataset

• PeMS04
• PeMS07
• PeMS08

Please refer to STSGCN (AAAI2020).

Model

Run

python main.py

Reference

Please cite our paper if you use the model in your own work:

@article{tang2022domain,
  title={Domain Adversarial Spatial-Temporal Network: A Transferable Framework for Short-term Traffic Forecasting across Cities},
  author={Tang, Yihong and Qu, Ao and Chow, Andy HF and Lam, William HK and Wong, SC and Ma, Wei},
  journal={arXiv preprint arXiv:2202.03630},
  year={2022}
}