sanitgupta / X-SEAIPR

Modelling the spread of COVID-19 in India

X-SEAIPR: Modelling the spread of COVID-19

This is the X-SEAIPR model. It attempts to model the spread of COVID-19 in India under different proposed interventions.

To clone the repository and enter its working directory:

>> git clone https://github.com/sanitgupta/X-SEAIPR.git
>> cd X-SEAIPR.git

To implement different interventions, including ramped up testing, lockdowns, and social distancing, make corresponding changes to Model.py

To run the model:

>> python Simulate.py

You will now have the results saved in .pkl files.

To generate the plots for each state/district after having run the model:

>> python PlotNew.py

You will now have the plots saved in the Plots folder.

If one wishes to run the model for a particular state (or for India but with newer data), instead of India, one has to replace the data files (time series case data, age-wise population data, transportation data) in the Data folder.

Further, changes can be made to Model.py and Simulate.py to run various other scenarios.

One can edit the setStateModels function in Model.py to change the lockdown dates, the dates for which contact reduction is done, the amount of contact reduction and the dates across which testing gets ramped up.

Also, one can edit the params dict inside the function to change the value of parameters like recovery rates, percent asymptomatic, etc.

To add more adaptive interventions, for example dynamic lockdowns where a state would shut down once a certain number of cases is reached, one must edit the dx function in Model.py.

To change the end date of the simulation, one needs to make the corresponding change in Simulate.py.

To use one's own estimates of testing fraction for different states/districts, one can edit the beta.json file in the Data folder.