Skip to content


Repository files navigation

When in Doubt: Neural Non-Parametric Uncertainty Quantification for Epidemic Forecasting

Paper Link:


First install Anaconda. The dependencies are listed in environment.yml file. Make sure you make changes to version of cudatoolkit if applicable.

Then run the following commands:

conda env create --prefix ./envs/epifnp --file environment.yml
source activate ./envs/epifnp

Directory structure

	- ILINet.csv -> wILI values for seasons 2003 to 2020 collected from flusight
- model_chkp -> stores intermediate model parameters while training
- models/ -> implementation of EpiFNP modules
- plots -> plots of predictions
- saves -> saves predictions for models as pkl files
- -> training script for EpiFNP
- -> inference of trained model
- -> Autoregressive inference using a trained model



python -y <test season> -w <week ahead> -a trans -n <experiment name> -e <max num. of epochs>

Or run to run all experiments.

Prediction plots will be saved in plots/Test<experiment name>.png and model in model_chkp folder.



python -y <test season> -w <week ahead> -a trans -n <experiment name>

for normal inference.


python -y <test season> -w <week ahead> -a trans -n <experiment name>

for auto-regressive inference. Note: Train and use a 1 week ahead model for AR inference.

The predictions and plots are saved in saves and plots respectively.