Exemple #1
0
def network_exp_with_open_up(
        transmissibility,
        sigma,
        symptomatic_rate,
        gamma,
        n,
        t,
        graph,
        npis,
        t_apply_npi,
        t_open_up,
        scale=1,
        no_npi_nodes=pickle.load(
            open("dataset/stable/contact_network/no_npi_nodes.pkl", "rb")),
        **kwargs):
    model1 = NetworkedSEIR(transmissibility=transmissibility,
                           sigma=sigma,
                           symptomatic_rate=symptomatic_rate,
                           gamma=gamma,
                           num_days=t,
                           scale=scale,
                           no_npi_nodes=no_npi_nodes)
    seir = model1.run_open_up_with_readjust(graph,
                                            npis,
                                            t_apply_npi,
                                            t_open_up,
                                            recovered=277,
                                            infected=9,
                                            **kwargs)
    return seir
def network_exp_with_seed(transmissibility,
                          sigma,
                          symptomatic_rate,
                          gamma,
                          n,
                          t,
                          graph,
                          scale=1):
    model1 = NetworkedSEIR(transmissibility=transmissibility,
                           sigma=sigma,
                           symptomatic_rate=symptomatic_rate,
                           gamma=gamma,
                           num_days=t,
                           scale=scale)
    seir = model1.run(graph)
    return seir
def network_exp_with_npi(transmissibility,
                         sigma,
                         symptomatic_rate,
                         gamma,
                         n,
                         t,
                         graph,
                         npis,
                         t_apply_npi,
                         scale=1,
                         no_npi_nodes=pickle.load(
                             open("no_npi_nodes.pkl", "rb")),
                         **kwargs):
    model1 = NetworkedSEIR(transmissibility=transmissibility,
                           sigma=sigma,
                           symptomatic_rate=symptomatic_rate,
                           gamma=gamma,
                           num_days=t,
                           scale=scale,
                           no_npi_nodes=no_npi_nodes)
    seir = model1.run_npi(graph, npis, t_apply_npi, **kwargs)
    return seir