Our work is already accepted by KDD2024 research track. The citation information will be updated when the official proceeding is online.
@article{dong2024heterogeneity,
title={Heterogeneity-Informed Meta-Parameter Learning for Spatiotemporal Time Series Forecasting},
author={Dong, Zheng and Jiang, Renhe and Gao, Haotian and Liu, Hangchen and Deng, Jinliang and Wen, Qingsong and Song, Xuan},
journal={arXiv preprint arXiv:2405.10800},
year={2024}
}
pytorch>=1.12
numpy
pandas
matplotlib
pyyaml
pickle
torchinfo
cd scripts/
python train.py -d <dataset> -g <gpu_id>
<dataset>
:
- METRLA
- PEMSBAY
- PEMS04
- PEMS07
- PEMS08