search

daveselinger / covid_rapid_test_simulation2

An update to our original rapid test simulation.
MIT License
0 stars 1 forks source link

Added infectionsDF(). Fixed random sampling. Added initial vaccinated & initial recovered. #21

Closed tedkyi closed 2 years ago

tedkyi commented 2 years ago

Created small helper simulation.infectionsDF() to output the infection spread data in a dataframe so it will be easy to pull into visualizations.

tedkyi commented 2 years ago

The places where we were selecting multiple people (e.g. interactions, initial infected) were incorrectly doing random sampling with replacement. I switched the code to use random.sample() to do sampling without replacement. Also added new parameters and new init code for people who are vaccinated and people who are recovered at the beginning of the simuation.