Source code for the paper, "HierST: A Unified Hierarchical Spatial-temporal Framework for COVID-19 Trend Forecasting", accepted by CIKM 2021.
- Install CUDA 10.1
- Run
setup_py.shto install necessary packages
cd src/
# pull data
python data_utils.py
# train models
python run_models.py --forecast_date 2021-10-24
# ensemble
python run_ensemble.py --forecast_date 2021-10-24
# check the output file, '../outputs/2021-10-24_forecasts.csv'