avsolatorio/DiffSTG

Code for DiffSTG

DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Models

Model Architecture

Run

1. requirements:

torch
easydict
nni

2. start training

python train.py

Cite

If you find this helpful, please cite our paper:

@misc{wen2023diffstg,
      title={DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Models}, 
      author={Haomin Wen and Youfang Lin and Yutong Xia and Huaiyu Wan and Qingsong Wen and Roger Zimmermann and Yuxuan Liang},
      year={2023},
      eprint={2301.13629},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}