Skip to content

square-coder/STGODE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

STGODE

This is an implementation of Spatial-Temporal Graph ODE Networks for Traffic Flow Forecasting

Run

python run_stode.py

Requirements

  • python 3.7
  • torch 1.7.0+cu101
  • torchdiffeq 0.2.2
  • fastdtw 0.3.4

Dataset

The datasets used in our paper are collected by the Caltrans Performance Measurement System(PeMS). Please refer to STSGCN (AAAI2020) for the download url.

Reference

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

@inproceedings{fang2021spatial,
  title={Spatial-Temporal Graph ODE Networks for Traffic Flow Forecasting},
  author={Fang, Zheng and Long, Qingqing and Song, Guojie and Xie, Kunqing},
  booktitle={Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery \& Data Mining},
  pages={364--373},
  year={2021}
}

About

Spatial-Temporal Graph ODE Neural Network

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages