From 5dfd768a7eb64a0b2ddce266616b3f46e480fff9 Mon Sep 17 00:00:00 2001 From: pjentsch <pjentsch@uwaterloo.ca> Date: Tue, 6 Jul 2021 23:00:52 -0400 Subject: [PATCH] parameter planes variance --- .../src/ABM/parameter_planes.jl | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/CovidAlertVaccinationModel/src/ABM/parameter_planes.jl b/CovidAlertVaccinationModel/src/ABM/parameter_planes.jl index 56f7b59..b6b9bf4 100644 --- a/CovidAlertVaccinationModel/src/ABM/parameter_planes.jl +++ b/CovidAlertVaccinationModel/src/ABM/parameter_planes.jl @@ -105,30 +105,30 @@ function plot_parameter_plane(input_fname) - mean_final_size(p) = mean(reduce(merge!,p.final_size_by_age)) + mean_final_size(p) = mean(reduce(merge!,p.final_size_by_age)); + std_final_size(p) = std(reduce(merge!,p.final_size_by_age)) base_outcome = mean_final_size(output_no_app) 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[age]) weighted_degree_map = [map(p -> mean_weighted_degree_change(p,i),output) for i in 1:3] - mean_notifications(p) = mean(reduce(merge!,p.daily_total_notifications)) - notifications_map = map(mean_notifications,output) + cs = cgrad(:blues) - datamaps = (weighted_degree_map..., final_size_map,notifications_map) + datamaps = (weighted_degree_map..., final_size_map,map(std_final_size,output_data)) fnames = [ "wdg_change_Y.pdf", "wdg_change_M.pdf", "wdg_change_O.pdf", "final_size_change.pdf", - "notifications.pdf" + "final_size_standard_dev.pdf" ] titles = [ "Average w. deg. of vaccinators minus average w. deg., Y", "Average w. deg. of vaccinators minus average w. deg., M", "Average w. deg. of vaccinators minus average w. deg., O", "Effect of notifications on tot. infections", - "total notifications" + "Standard deviation from the mean of total infection size" ] for (fname,title,datamap) in zip(fnames,titles,datamaps) p = heatmap(var_ranges[1],var_ranges[2],datamap; title, xlabel = vars[1], ylabel = vars[2], seriescolor=cs, size = (600,400)) -- GitLab