def graph(self): if os.path.isfile(fpath_db): os.remove(fpath_db) te = TrajectoryEnsemble(fpath_db) if te.is_db_empty: te.set_pragma_memoize_group_ids(True) te.add_trajectories([ (Simulation(). add([ SARSQuarantineIntervention( SEQIHRModel('sars', beta=0.80, alpha_n=0.75, alpha_q=0.40, delta_n=0.01, delta_h=0.03, mu=0.01, chi=0.01, phi=0.20, rho=0.75, solver=ODESolver()), chi=0.99, i=int(intervention_onset) ), Group(m=95000, attr={ 'sars': 's' }), Group(m= 5000, attr={ 'sars': 'e' }) ]) ) for intervention_onset in TN(30,120, 75,100, 5) ]) te.set_group_names(group_names) sys.stdout = open('out.dat', 'w') s.run(400) sys.stdout.close() with open('out.dat') as file: self.data = file.readlines() self.data = str(self.data) apple = self.data.replace('\\n', '<br>') # print(apple) te.plot_mass_locus_line ((1200,300), os.path.join(os.path.dirname(__file__), 'out', 'plot-line.png'), col_scheme='tableau10', opacity_min=0.35) te.plot_mass_locus_line_aggr((1200,300), os.path.join(os.path.dirname(__file__), 'out', 'plot-ci.png'), col_scheme='tableau10') fpath_diag = os.path.join(os.path.dirname(__file__), 'out', 'sim-04.diag') fpath_pdf = os.path.join(os.path.dirname(__file__), 'out', 'sim-04.pdf') te.traj[2].sim.gen_diagram(fpath_diag, fpath_pdf)
# ---------------------------------------------------------------------------------------------------------------------- if os.path.isfile(fpath_db): os.remove(fpath_db) te = TrajectoryEnsemble(fpath_db) # te = TrajectoryEnsemble(fpath_db, cluster_inf=ClusterInf(address='auto')) # te = TrajectoryEnsemble(fpath_db, cluster_inf=ClusterInf(num_cpus=6, memory=500*1024*1024, object_store_memory=500*1024*1024, include_webui=True)) if te.is_db_empty: te.set_pragma_memoize_group_ids(True) te.add_trajectories([ Trajectory( (Simulation(). add([ SIRModel('flu', beta=0.10, gamma=0.05, solver=MCSolver()), SIRModel('flu', beta=0.50, gamma=U(0.01, 0.15), i=[int(5 + TN(0,50, 5,10)), 50], solver=MCSolver()), FluProcess(i=[50,0], p_max=None, a=3.0, scale=flu_proc_scale), Group(m=950, attr={ 'flu': 's' }), Group(m= 50, attr={ 'flu': 'i' }) ]) ) ) for flu_proc_scale in U(1,5, 1) ]) te.set_group_names(group_names) te.run(120) # Visualize: te.plot_mass_locus_line ((1200,300), os.path.join(os.path.dirname(__file__), 'out', '_plot-line.png'), opacity_min=0.2) te.plot_mass_locus_line_aggr((1200,300), os.path.join(os.path.dirname(__file__), 'out', '_plot-iqr.png'))