Commit 56242722 authored by Peter Jentsch's avatar Peter Jentsch
Browse files

assortavity heatmaps

parent f77f5822
const samples = 15
const samples = 25
const univariate_path = "CovidAlertVaccinationModel/plots/univariate/"
EN_begin_settings(p) = (
all_season = Intervals.Interval(0,p.sim_length),
......@@ -55,12 +55,12 @@ end
import AxisKeys
function multivariate_simulations()
len = 8
len = 10
default_parameters = get_app_parameters()
app_simulations = (
(:η, range(0.0, 5.0; length = len)),
(:ω_en, range(0.0, 5e-1; length = len)),
(:η,vcat([-0.5,-0.25],range(0.0, 3.0; length = len))),
(:ω_en,vcat([-0.1,-0.05], range(0.0, 3e-1; length = len))),
(:EN_intervals, EN_begin_settings(default_parameters))
# (:notification_threshold, (1:len)),
)
......@@ -119,15 +119,14 @@ function plot_parameter_plane(input_fname)
final_size_map = map(x-> (mean_final_size(x) - base_outcome),output)
mean_weighted_degree_change(p,age) = mean(p.avg_weighted_degree_of_vaccinators[age])-mean(p.avg_weighted_degree_of_vaccinators_no_EN[age])
weighted_degree_map = [map(p -> mean_weighted_degree_change(p,i),output) for i in 1:3]
mean_join_11_count(p) = mean(reduce(merge!,p.vac_join_counts[end-30:end]))
mean_join_01_count(p) = mean(p.vac_join_01_counts[end])
# mean_join_00_count(p) = mean(p.vac_join_00_counts[end])
join_count_maps = [
map(x-> (mean_join_11_count(x) - mean_join_11_count(output_no_app)),output),
map(x-> (mean_join_01_count(x) - mean_join_11_count(output_no_app)),output),
map(x-> (mean(x.assortativity_at_end)),output),
map(x-> (mean(x.vac_join_11_counts_at_end)),output),
map(x-> (mean(x.vac_join_01_counts_at_end)),output),
map(x-> (mean(x.vac_join_00_counts_at_end)),output),
# map(x-> (mean_join_00_count(x)),output),
]
for en_interval in var_ranges[3]
......@@ -141,7 +140,8 @@ function plot_parameter_plane(input_fname)
"wdg_change_O.pdf",
"final_size_change.pdf",
"final_size_standard_dev.pdf",
"join_counts_imm.pdf",
"assortativity.pdf",
"join_11_counts.pdf",
"join_01_counts.pdf",
"join_00_counts.pdf",
]
......@@ -151,8 +151,10 @@ function plot_parameter_plane(input_fname)
"Change in w. deg. of vaccinators with EN",
"Effect of EN on tot. infections",
"Standard deviation from the mean of total infection size",
"Join counts (J_11 + J_00)",
"Weighted assortativity",
"Join counts (J_11)",
"Join counts (J_01)",
"Join counts (J_00)",
]
for (fname,title,datamap) in zip(fnames,titles,datamaps)
p = heatmap(var_ranges[1],var_ranges[2],transpose(datamap(EN_intervals = en_interval)); title, xlabel = vars[1], ylabel = vars[2], seriescolor=cs,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment