mila-iqia / COVI-AgentSim

Covid-19 spread simulator with human mobility and intervention modeling.
Other
20 stars 8 forks source link

Epi metrics #62

Closed mweiss17 closed 4 years ago

mweiss17 commented 4 years ago

Description

@AndrewRWilliams, so here's whatcha wanna do. You're going to want to setup a "test" dataset. The way you do this is by running twice: python src/covid19sim/run.py init_fraction_sick=0.05 INTERVENTION_DAY=1 simulation_days=10 tune=True intervention=heuristicv3

This should yield two output directories in output/ containing pickle files which contain tracker data. You're going to need to move these two seeds into a new directory called results/test/heuristicv3/. You may also wish to generate some more data with a different intervention, may I recommend intervention=bdt1, and then put those into results/test/bdt1/<name_of_folder_in_output>

Next, you run python src/covid19sim/plotting/main.py plot=qalys path=results/test. This will load the data and run src/covid19sim/plotting/plot_qalys.py. Now you get to compute some qalys in there.

If you need to generate additional data (not a bad idea if you add more stuff to tracker) then delete results/test/cache.pkl which is generated when you run the main.py plotscript. If you need additional information for the qalys and can't figure out how to extract it, just let me know and I can help grab it.