src/iohmm_evac/bootstrap/__init__.py

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# SPDX-License-Identifier: AGPL-3.0-only
# Copyright (C) 2026 SWGY, Inc.
"""Parametric bootstrap on the IO-HMM-fitted population (Build 4).

This subpackage resamples households with replacement, refits the IO-HMM on
each replicate, and runs simulations under a sweep of warning-time shifts so
the chapter can plot policy outcomes (Fig. 6) with parametric-bootstrap
uncertainty bands.

See ``docs/bootstrap.md`` for the procedure and the rationale for warm
starting + deterministic simulation seeds.
"""

from __future__ import annotations

from iohmm_evac.bootstrap.aggregate import BandResult, compute_bands
from iohmm_evac.bootstrap.resample import (
    ResampledFitData,
    index_fit_data,
    resample_indices,
)
from iohmm_evac.bootstrap.runner import (
    BootstrapFit,
    load_bootstrap_fits,
    run_bootstrap_fits,
)
from iohmm_evac.bootstrap.shift_sweep import (
    ShiftSweepResult,
    SweepRow,
    load_sweep_result,
    run_shift_sweep,
)

__all__ = [
    "BandResult",
    "BootstrapFit",
    "ResampledFitData",
    "ShiftSweepResult",
    "SweepRow",
    "compute_bands",
    "index_fit_data",
    "load_bootstrap_fits",
    "load_sweep_result",
    "resample_indices",
    "run_bootstrap_fits",
    "run_shift_sweep",
]