laser.measles.components.base_tracker

base_tracker

Base tracker component for laser.measles models.

BaseTracker(model, params=None)

Bases: BaseComponent, ABC

Abstract base class for tracker components.

Examples:

from laser.measles.scenarios.synthetic import single_patch_scenario
from laser.measles.biweekly import BiweeklyModel, BiweeklyParams
from laser.measles.biweekly import components
from laser.measles import create_component

scenario = single_patch_scenario(population=100_000, mcv1_coverage=0.85)
params = BiweeklyParams(num_ticks=52, seed=42, start_time="2000-01")
model = BiweeklyModel(scenario, params)
model.add_component(create_component(components.StateTracker, components.StateTrackerParams()))

get_data()

Get all tracked data.

initialize(model) abstractmethod

Initialize tracker with model.

reset()

Reset tracked data.

should_track(tick)

Check if we should track data at this time step.

update(model, tick) abstractmethod

Update tracker with current model state.

BaseTrackerParams

Bases: BaseModel

Common parameters for tracker components.

Examples:

from laser.measles.biweekly.components.tracker_state import StateTrackerParams

params = StateTrackerParams(track_states=True, output_frequency=1)