This repository contains code for the paper Evaluating COVID-19 vaccine allocation policies using Bayesian m-top exploration
- envs: The environments used for running the algorithms
- mab: Multi-armed bandits
- loggers: Logging for bandits and top-m algorithms
- resources: Vaccine supply based on weekly deliveries
The use of the STRIDE environment requires the vaccine extension and pybind11 wrapper implemented in the STRIDE fork. Additionally, the synthetic population data for Belgium is necessary to simulate the entire 11 million population.
The full mapping of arm numbers to vaccine strategies can be found here.
Run simulations for a single arm on STRIDE
python mab/play_arm.py --help
Run simulations for a bandit on STRIDE
python mab/play_bandit.py --help
Run simulations for a bandit on the ground truth
python mab/play_bandit_gt.py --help