Skip to content

Latest commit

 

History

History
38 lines (21 loc) · 1.81 KB

README.md

File metadata and controls

38 lines (21 loc) · 1.81 KB

PatchSim

Code for simulating the metapopulation SEIR model. Sample network for US national simulation included. DOI

A preliminary description of this model appeared in IEEE ICHI 2017: https://ieeexplore.ieee.org/document/8031141/

Journal version: https://journals.plos.org/ploscompbiol/article?id=10.1371/journal.pcbi.1007111 Cite as: Venkatramanan S, Chen J, Fadikar A, Gupta S, Higdon D, Lewis B, Marathe M, Mortveit H, Vullikanti A. Optimizing spatial allocation of seasonal influenza vaccine under temporal constraints. PLoS computational biology. 2019 Sep 16;15(9):e1007111.

Also used in upcoming paper: Venkatramanan S, Sadilek A, Fadikar A, et al. Forecasting influenza activity using machine-learned mobility map, to appear in Nature Communications.

Documentation

The description of the model and a software manual can be found inside the doc/ folder.

Dependencies

PatchSim is compatible with Python 3.5+. It requires numpy and pandas.

Testing

Please use the different test*.py scripts to test the different functionalities of PatchSim. For more details on these features please check the software manual.

US National simulation

To test the US national simulation, extract the zip file test/US_county.zip and run "python test_det_US.py" from inside test/ folder.

Using GPU (with CuPy) for core simulation

Install CuPy version compatible with your CUDA version. For example: pip install cupy-cuda11x

Before importing PatchSim set PATCHSIM_USE_GPU environment variable. For example:

  • shell: export PATCHSIM_USE_GPU=1, before running PatchSim script
  • jupyter notebook: %env PATCHSIM_USE_GPU=1, before PatchSim import
  • python script: import os; os.environ['PATCHSIM_USE_GPU'] = '1', before PatchSim import