- numpy
- scipy
- matplotlib
You can install these using pip:
pip install numpy scipy matplotlib
- Run the simulate_diff.py script to perform the simulation and plot the results:
- Create the data by running
python3 src/data.py
, this creates adata.txt
file which you can then read from. - Run
python3 src/simulate_diff.py
This will display a plot showing the number of susceptible, infected, and recovered individuals over time.
This is just a simple project I dared to work on after several days of Tyhpoid fever lol. Submit a PR if you're inspired enough.
To learn more about the SIR model and disease modeling in general, you might find the following resources helpful: