WadhwaniAI / covid-modelling

Repo for modelling the spread of novel Covid-19
MIT License
5 stars 0 forks source link

semantics of the output #20

Closed smerugu closed 3 years ago

smerugu commented 4 years ago

https://github.com/WadhwaniAI/covid-modelling/blob/160c4d17da7a77b6b8aeaa8b967b21db30d52429/models/seir/seir_testing.py#L165

I think we should return all the compartment counts here and then have another predict function that maps it to the typical expected outputs (hospitalized, confirmed, total infected) with the right definitions. We are not including mild in hospitalized and confirmed typically doesn't have unknown infectious

sansiddh commented 3 years ago

Done. This is how it is done now