The dataset of AQI-36 and PEMS-BAY can be downloaded by https://github.com/LMZZML/PriSTI and the

The P12 dataset can be download by python download.py physio

training and imputation for the AQI dataset

exe_pm25.py

training and imputation for the P12 dataset

exe_physio.py

training and imputation for the PEMS-BAY dataset

exe_pemsbay.py

Acknowledgements A part of the codes is based on CSDI https://github.com/ermongroup/CSDI

I would like to express my sincere gratitude to the following individuals and repositories, as their work has greatly inspired and contributed to this project: @inproceedings{tashiro2021csdi,
title={CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputation},
author={Tashiro, Yusuke and Song, Jiaming and Song, Yang and Ermon, Stefano}, booktitle={Advances in Neural Information Processing Systems},
year={2021} }

Also, I would like to express my gratitude to the following individuals, repositories, and models that have inspired to the imputation task:

@article{liu2023pristi,
title={PriSTI: A Conditional Diffusion Framework for Spatiotemporal Imputation},
author={Liu, Mingzhe and Huang, Han and Feng, Hao and Sun, Leilei and Du, Bowen and Fu, Yanjie},
journal={arXiv preprint arXiv:2302.09746}, year={2023} }

@article{lopez2022diffusion, title={Diffusion-based time series imputation and forecasting with structured state space models}, author={Lopez Alcaraz, Juan Miguel and Strodthoff, Nils},
journal={arXiv e-prints}, pages={arXiv--2208},
year={2022} }

@article{cini2021filling,
title={Filling the g_ap_s: Multivariate time series imputation by graph neural networks}, author={Cini, Andrea and Marisca, Ivan and Alippi, Cesare},
journal={arXiv preprint arXiv:2108.00298},
year={2021} }

@article{du2023saits, title = {{SAITS: Self-Attention-based Imputation for Time Series}}, journal = {Expert Systems with Applications}, volume = {219}, pages = {119619}, year = {2023}, issn = {0957-4174}, doi = {10.1016/j.eswa.2023.119619}, url = {https://arxiv.org/abs/2202.08516}, author = {Wenjie Du and David Cote and Yan Liu}, }