In [1]:

    

from dzdy import *
import matplotlib.pyplot as plt


    

In [2]:

    

da = Director()
da.load_pc('scripts/pCare.txt')
da.load_dc('scripts/CareSeeking.txt')


    

In [3]:

    

mod = da.new_mc('Screening1', 'ABM', tar_pc='pSeeking', tar_dc='CareSeeking')
mod.set_observations(transitions=['Fail', 'Cure'], behaviours=['screen'])

mod = da.new_mc('Screening2', 'ABM', tar_pc='pSeeking', tar_dc='CareSeeking')
mod.add_behaviour('screen', be_type='NerfDecision', s_src = 'Sick', t_tar = 'Cure', prob=0.5)
mod.set_observations(transitions=['Fail', 'Cure'], behaviours=['screen'])

mod = da.new_mc('Screening3', 'ABM', tar_pc='pSeeking', tar_dc='CareSeeking')
mod.add_behaviour('screen', be_type='BuffDecision', s_src = 'Sick', t_tar = 'Fail', prob=0.5)
mod.set_observations(transitions=['Fail', 'Cure'], behaviours=['screen'])


    

In [5]:

    

a1, out =  da.simulate('Screening1', y0= {'Latent': 1000}, to=20)
out.plot()

a2, out =  da.simulate('Screening2', y0= {'Latent': 1000}, to=20)
out.plot()

a1, out =  da.simulate('Screening3', y0= {'Latent': 1000}, to=20)
out.plot()
plt.show()