comp_plot function

Plot Compartment Diagram for Epidemic Models