Skip to content

Commit

Permalink
base values on pipeline types
Browse files Browse the repository at this point in the history
  • Loading branch information
SamuelBrand1 committed Dec 19, 2024
1 parent e894504 commit 73e80c1
Showing 1 changed file with 16 additions and 2 deletions.
18 changes: 16 additions & 2 deletions pipeline/scripts/create_postprocessing_dataframes.jl
Original file line number Diff line number Diff line change
@@ -1,10 +1,24 @@
using EpiAwarePipeline, EpiAware, JLD2, DrWatson, DataFramesMeta, CSV, MCMCChains

pipelinetypes = [
MeasuresOutbreakPipeline,
SmoothOutbreakPipeline,
SmoothEndemicPipeline,
RoughEndemicPipeline
]
## Define scenarios
scenarios = ["measures_outbreak", "smooth_outbreak", "smooth_endemic", "rough_endemic"]

scenarios = pipelinetypes .|> pipetype -> pipetype().prefix

## Define true GI means
true_gi_means = [2.0, 10.0, 20.0]
# Errors if not the same for all pipeline types
true_gi_means = map(pipelinetypes) do pipetype
make_gi_params(pipetype())["gi_means"]
end |>
ensemble_gi_means -> all([gi_means == ensemble_gi_means[1]
for gi_means in ensemble_gi_means]) ?
ensemble_gi_means[1] :
error("GI means are not the same")

if !isfile(plotsdir("plotting_data/predictions.csv"))
@info "Prediction dataframe does not exist, generating now"
Expand Down

0 comments on commit 73e80c1

Please sign in to comment.