from Simulator import Simulator from States import S_Start, S_DeadCancer, S_DeadAge, S_DeadChemo from numpy import average import ConundrumModel as Model s= Simulator(iters=10000, age=60, ca125=True, model=Model) s.run() print average([len(p.states) for p in s.patients]) s.clear_survival_curves() s.add_survival_curve(start_states=[S_Start], end_states=[S_DeadCancer]) s.add_survival_curve(start_states=[S_Start], end_states=[S_DeadChemo]) s.add_survival_curve(start_states=[S_Start], end_states=[S_DeadAge]) s.draw_survival_curves(name="conundrum.png")