Source code for sbmlsim.examples.experiments.covid.simulate

"""
Run COVID-19 model experiments.
"""
from pathlib import Path

from sbmlsim.examples.experiments.covid.experiments import (
    Bertozzi2020,
    Carcione2020,
    Cuadros2020,
)
from sbmlsim.experiment import ExperimentRunner
from sbmlsim.report.experiment_report import ExperimentReport, ReportResults
from sbmlsim.simulator.simulation_ray import SimulatorParallel


[docs]def run_covid_experiments(output_path: Path) -> None: """Run covid simulation experiments.""" base_path = Path(__file__).parent runner = ExperimentRunner( [ Cuadros2020, # Bertozzi2020, Carcione2020, ], simulator=SimulatorParallel(), base_path=base_path, data_path=base_path, ) results = runner.run_experiments( output_path=output_path, show_figures=True, reduced_selections=False, ) report_results = ReportResults() for exp_result in results: report_results.add_experiment_result(exp_result=exp_result) report = ExperimentReport(report_results) report.create_report(output_path=output_path)
if __name__ == "__main__": run_covid_experiments(Path(__file__).parent / "results")