search

epirecipes / sir-julia

Various implementations of the classical SIR model in Julia
MIT License
196 stars 39 forks source link

InfiniteArrays.jl example #42

Closed slwu89 closed 2 years ago

slwu89 commented 2 years ago

Hi @sdwfrost, opening a PR to work on #39. Looking forward to hearing your feedback, right now the example uses DifferentialEquations.jl with a struct that stores the infinite array to track the number of infectious persons who have infected 1,2,... other persons.

sdwfrost commented 2 years ago

Thanks @slwu89! I'll add in some more details of the model later but am merging for now (if you calculate the next generation matrix of the model and compute the eigenvector of the NGM, it shows the geometric series nicely). The closest kind of approach to this that I've seen is in @kerguler sPop C library (now sPop2) that dynamically keeps different age classes in a data structure.

slwu89 commented 2 years ago

That's very interesting but makes a lot of sense intuitively, and a fun mathematical exercise to work out later! Thanks for the sPop2 link, I played with sPop 1, didn't know it had such a significant upgrade lately.