# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: Python 3
#     language: python
#     name: python3
# ---

from autumn.tools.project import get_project
from matplotlib import pyplot
from autumn.tools.plots.utils import REF_DATE
from autumn.tools.calibration.targets import get_target_series
import pandas as pd
from autumn.tools.utils.pretty import pretty_print

project = get_project("tuberculosis", "marshall-islands")

model = project.run_baseline_model(project.param_set.baseline)
derived_df = model.get_derived_outputs_df()

output = "prevalence_infectiousXlocation_majuro"
target = (2018, 1366)

# +
# pretty_print(custom_params)
# -

fig = pyplot.figure(figsize=(12, 8))
pyplot.style.use("ggplot")
axis = fig.add_subplot()
axis = derived_df[output].plot()
axis.scatter(target[0], target[1], color="black")



StarcoderdataJupyterScriptsDedupFiltered