[docs]defmake_tb_nut():# --------- People ----------n_agents=10000extra_states=[ss.FloatArr('SES',default=ss.bernoulli(p=0.3)),# SES example: ~30% get 0, ~70% get 1 (TODO)]pop=ss.People(n_agents=n_agents,extra_states=extra_states)# ------- TB disease --------# Disease parameterstb_pars=dict(beta=ss.peryear(0.01),init_prev=0.25,)# Initializetb=mtb.TB(tb_pars)# ---------- Malnutrition --------nut_pars=dict()nut=mtb.Malnutrition(nut_pars)# -------- Network ---------# Network parametersnet_pars=dict(n_contacts=ss.poisson(lam=5),dur=0,# End after one timestep)# Initialize a random networknet=ss.RandomNet(net_pars)# Add demographicsdems=[ss.Pregnancy(pars=dict(fertility_rate=15)),# Per 1,000 peopless.Deaths(pars=dict(death_rate=10)),# Per 1,000 people]# Connectorcn_pars=dict()cn=mtb.TB_Nutrition_Connector(cn_pars)# -------- simulation -------# define simulation parameterssim_pars=dict(dt=ss.days(7),start=ss.date('1990-01-01'),stop=ss.date('2010-12-31'),)# initialize the simulationsim=ss.Sim(people=pop,networks=net,diseases=[tb,nut],pars=sim_pars,demographics=dems,connectors=cn)sim.pars.verbose=0.1# Print status every 10% of simulationreturnsim