gridemic is a python package for stochastic agent-based epidemic models on a square grid with long-range interactions.
In addition to the disease spread, gridemic also simulates the testing & contact tracing simultaneously.
pip install gridemic
The notebooks in examples/ demonstrates basic functionality of gridemic. The video below was generated by running examples/SEIIRT5.ipynb.
SEIIRT5.mp4
D. Scarselli, N. B. Budanur, M. Timme, and B. Hof "Discontinuous epidemic transition due to limited testing", Nature Communications volume 12, Article number: 2586 (2021). arXiv:2006.08005.