-
Notifications
You must be signed in to change notification settings - Fork 0
/
driver_vaccine_scenario.py
108 lines (54 loc) · 2.76 KB
/
driver_vaccine_scenario.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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# -*- coding: utf-8 -*-
"""
Created on Tue Mar 22 10:21:09 2022
@author: Nicolas Rebuli
Test script for the development of the vaccine dynamics implemented
inplimented in the model.
"""
#%% SETUP Modules
# Modules
from multiprocessing import Pool
import src.vaccinations.vaccination as vax
# Import vaccination parameters
# from src.vaccination_scenarios.test_parameters import vax_parameters
# from src.vaccination_scenarios.test_1_vax_distributed_no_effect import vax_parameters
# from src.vaccination_scenarios.test_2_vax_means_no_infections import vax_parameters
# from src.vaccination_scenarios.test_3_vax_means_infection_half_as_likely import vax_parameters
# from src.vaccination_scenarios.test_4_vax_means_no_transmission import vax_parameters
# from src.vaccination_scenarios.test_5_vax_means_half_transmission import vax_parameters
# from src.vaccination_scenarios.test_6_vax_means_no_detection import vax_parameters
# from src.vaccination_scenarios.test_7_vax_means_half_detection import vax_parameters
# from src.vaccination_scenarios.test_8_vax_means_no_infectious_period import vax_parameters
# from src.vaccination_scenarios.test_9_vax_means_half_infectious_period import vax_parameters
# from src.vaccination_scenarios.test_10_vax_means_all_effects import vax_parameters
# from src.vaccination_scenarios.test_11_vax_means_no_effects import vax_parameters
# from src.vaccination_scenarios.test_12_vax_means_no_rectal import vax_parameters
# from src.vaccination_scenarios.test_13_vax_means_no_pharyngeal import vax_parameters
# from src.vaccination_scenarios.test_14_vax_means_no_urethral import vax_parameters
# from src.vaccination_scenarios.test_15_vax_means_all_effects_at_debut import vax_parameters
# from src.vaccination_scenarios.test_16_vax_means_all_effects_all_cohorts import vax_parameters
# from src.vaccination_scenarios.test_17_vax_everyone import vax_parameters
# from src.vaccination_scenarios.vax_scenario_0_baseline import vax_parameters
from src.vaccination_scenarios.vax_example import vax_parameters
#%% RUN SIMULATIONS
# Set parameters
run_mode = 'serial'
n_cores = min(16, vax_parameters['n_sims'])
sim_run = list(range(0, vax_parameters['n_sims']))
# If running in serial
if run_mode == 'serial':
# Run all the sims
for sim_no in sim_run:
vax.run_vaccine_scenario(vax_parameters, sim_no, run_mode = 'serial')
# If running in parallel
if run_mode == 'parallel':
# Define worker for running the simulation
def worker(it):
vax.run_vaccine_scenario(vax_parameters, it, run_mode = 'parallel')
# Define pool handler
def pool_handler():
p = Pool(n_cores)
p.map(worker, sim_run)
# Batch scripts
if __name__ == '__main__':
pool_handler()