stisim.networks module

Define sexual network for STI transmission.

Overview:

• Risk groups: agents are randomly assigned into one of 3 main risk groups:

  • 0 = marry and remain married to a single partner throughout their lifetime

  • 1 = marry and then divorce or who have concurrent partner(s) during their marriage

  • 2 = never marry

• In addition, a proportion of each of the groups above engages in sex work.

class stisim.networks.StructuredSexual(pars=None, key_dict=None, condom_data=None, name=None, **kwargs)

Bases: SexualNetwork

Structured sexual network

static process_condom_data(condom_data)

get_age_risk_pars(uids, par)

init_pre(sim)

Perform initialization steps

This method is called once, as part of initializing a Sim. Note: after initialization, initialized=False until init_vals() is called (which is after distributions are initialized).

init_results()

Store network length by default

init_post()

Initialize the values of the states; the last step of initialization

set_network_states(upper_age=None)

Many network states depend on properties of people – e.g. MSM depends on being male, age of debut varies by sex and over time, and participation rates vary by age. Each time states are dynamically grown, this function should be called to set the network states that depend on other states.

over_debut()

set_risk_groups(upper_age=None)

Assign each person to a risk group

set_concurrency(upper_age=None)

Assign each person a preferred number of simultaneous partners

set_sex_work(upper_age=None)

set_debut(upper_age=None)

match_pairs(ppl)

Match pairs by age

add_pairs(ti=None)

Add pairs

match_sex_workers(ppl)

Match sex workers to clients

end_pairs()

update_results()

Store the number of edges in the network

net_beta(disease_beta=None, uids=None, disease=None)

Calculate the beta for the given disease and network

set_condom_use()

Set condom use

count_partners()

Count the number of partners each person has had over the past 3/12 months

step()

Define how the module updates over time – the key part of Starsim!!

class stisim.networks.FastStructuredSexual(**kwargs)

Bases: StructuredSexual

match_pairs(ppl)

Match pairs by age, using sorting rather than the linear sum assignment

class stisim.networks.AgeMatchedMSM(**kwargs)

Bases: StructuredSexual

match_pairs(ppl)

Match males by age using sorting

class stisim.networks.AgeApproxMSM(**kwargs)

Bases: StructuredSexual

match_pairs(ppl)

Match