# Model Build """ ___________ | | S --> I --> H --> R """ Susceptible = epi.comps.Susceptible(0, R_0, gamma, N) Infected = epi.comps.Infected(1, N, R_0=R_0, gamma=gamma, p_recovery=p_rec, recovery_rate=rec_rate, p_hospitalized=p_hos, hospital_rate=hos_rate) Hospitalized = epi.comps.Hospitalized(2, hos_rate, p_hos, recovery_rate=hos_stay, p_recovery=p_hos_to_rec) Recovered = epi.comps.Recovered(3, p_from_inf=p_rec, from_inf_rate=rec_rate, p_from_hos=p_hos_to_rec, from_hos_rate=hos_stay) # Model Compiler Compiled_Model = epi.models.Model(N(0), layers=[Susceptible, Infected, Hospitalized, Recovered], layer_names=['Susceptible', 'Infected', 'Hospitalized', 'Recovered'], layer_map=[[Infected], [Hospitalized, Recovered], [Recovered], []])