start2020 / Multi-View_TRGRU

Multi-View TRGRU: Transformer based Spatiotemporal Model for Short-Term Metro Origin-Destination Matrix Prediction

This is a TensorFlow implementation of Multi-View TRGRU.

Requirement

• numpy==1.18.1

• pandas==1.0.3

• tensorflow==2.1.0

• tensorflow-gpu==2.1.0

Citation

If you find this repository useful in your research, please cite the following paper:

@article{DBLP:journals/corr/abs-2108-03900,
  author    = {Jiexia Ye and
               Furong Zheng and
               Juanjuan Zhao and
               Kejiang Ye and
               Chengzhong Xu},
  title     = {Multi-View {TRGRU:} Transformer based Spatiotemporal Model for Short-Term
               Metro Origin-Destination Matrix Prediction},
  journal   = {arXiv},
  year      = {2021}
}