fpsim.analyzers module

Specify the core analyzers available in FPsim. Other analyzers can be defined by the user by inheriting from these classes.

class fpsim.analyzers.Analyzer(label=None)

Bases: sciris.sc_utils.prettyobj

Base class for analyzers. Based on the Intervention class. Analyzers are used to provide more detailed information about a simulation than is available by default – for example, pulling states out of sim.people on a particular timestep before it gets updated in the next timestep.

To retrieve a particular analyzer from a sim, use sim.get_analyzer().

Parameters

label (str) – a label for the Analyzer (used for ease of identification)

initialize(sim=None)

Initialize the analyzer, e.g. convert date strings to integers.

finalize(sim=None)

Finalize analyzer

This method is run once as part of sim.finalize() enabling the analyzer to perform any final operations after the simulation is complete (e.g. rescaling)

apply(sim)

Apply analyzer at each time point. The analyzer has full access to the sim object, and typically stores data/results in itself. This is the core method which each analyzer object needs to implement.

Parameters

sim – the Sim instance

to_json()

Return JSON-compatible representation

Custom classes can’t be directly represented in JSON. This method is a one-way export to produce a JSON-compatible representation of the intervention. This method will attempt to JSONify each attribute of the intervention, skipping any that fail.

Returns

JSON-serializable representation

class fpsim.analyzers.snapshot(timesteps, *args, die=True, **kwargs)

Bases: fpsim.analyzers.Analyzer

Analyzer that takes a “snapshot” of the sim.people array at specified points in time, and saves them to itself.

Parameters

• timesteps (list) – list of timesteps on which to take the snapshot

• args (list) – additional timestep(s)

• die (bool) – whether or not to raise an exception if a date is not found (default true)

• kwargs (dict) – passed to Analyzer()

Example:

sim = cv.Sim(analyzers=fps.snapshot('2020-04-04', '2020-04-14'))
sim.run()
snapshot = sim.pars['analyzers'][0]
people = snapshot.snapshots[0]

apply(sim)

Apply snapshot at each timestep listed in timesteps and save result at snapshot[str(timestep)]

class fpsim.analyzers.timeseries_recorder

Bases: fpsim.analyzers.Analyzer

Record every attribute in people as a timeseries.

self.i

The list of timesteps (ie, 0 to 261 steps).

self.t

The time elapsed in years given how many timesteps have passed (ie, 25.75 years).

self.y

The calendar year of timestep (ie, 1975.75).

self.keys

A list of people states excluding ‘dobs’.

self.data

A dictionary where self.data[state][timestep] is the mean of the state at that timestep.

initialize(sim)

Initializes self.keys from sim.people

apply(sim)

Applies recorder at each timestep

plot(x='y', fig_args=None, pl_args=None)

Plots time series of each state as a line graph

class fpsim.analyzers.age_pyramids(bins=None)

Bases: fpsim.analyzers.Analyzer

Records age pyramids for each timestep.

self.bins

A list of ages, default is a sequence from 0 to max_age + 1.

self.data

A matrix of shape (number of timesteps, number of bins - 1) containing age pyramid data.

initialize(sim)

Initializes bins and data with proper shapes

apply(sim)

Records histogram of ages of all alive individuals at a timestep such that self.data[timestep] = list of proportions where index signifies age

plot()

Plots self.data as 2D pyramid plot

plot3d()

Plots self.data as 3D pyramid plot

class fpsim.analyzers.SimVerbose(pars=None, mother_ids=False)

Bases: fpsim.sim.Sim

log_daily_totals()

Logs data for total_results and events at each timestep.

Output:

self.total_results::dict

Dictionary of all individual results formatted as {timestep: attribute: [values]} keys correspond to fpsim.defaults debug_states

self.events::dict

Dictionary of events correponding to self.channels formatted as {timestep: channel: [indices]}.

save_daily_totals()

At the end of sim run, stores total_results as either a json or feather file.

Inputs

self.to_file::bool

If True, writes results to file

self.to_feather::bool

If True, writes results to feather file If False, writes results to json file

Outputs:

Either a json file at “sim_output/total_results.json” or a feather file for each state at “sim_output/{state}_state”

story(index)

Prints a story of all major events in an individual’s life based on calculated SimVerbose channels, base Sim channels, and statistics calculated within the function such as year of birth of individual.

Inputs:

index::int:

index of the individual, must be less than population

Outputs:

printed display of each major event in the individual’s life

class fpsim.analyzers.ExperimentVerbose(pars=None, flags=None, label=None)

Bases: fpsim.experiment.Experiment

run_model(pars=None, mother_ids=False)

Create the sim and run the model, saving total results and individual events in the process