You can find the notebooks accompanying the medium articles on infectious disease modelling here They are named part_one.ipynb, etc.