Skip to content

Commit 679af21

Browse files
committed
Update make_model_priors.jl
1 parent 450b02a commit 679af21

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

pipeline/src/constructors/make_model_priors.jl

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,16 +8,18 @@ parameters. This is the default method.
88
# Returns
99
A dictionary containing the following prior distributions:
1010
- `"transformed_process_init_prior"`: A normal distribution with mean 0.0 and
11-
standard deviation 0.25.
12-
- `"std_prior"`: A half-normal distribution with standard deviation 0.25.
13-
- `"damp_param_prior"`: A beta distribution with shape parameters 0.5 and 0.5.
11+
standard deviation 0.025.
12+
- `"std_prior"`: A half-normal distribution with standard deviation 0.025.
13+
- `"damp_param_prior"`: A beta distribution with shape parameters 1 and 9.
14+
- `"log_I0_prior"`: A normal distribution with mean log(100.0) and standard
15+
deviation 1e-1.
1416
1517
"""
1618
function make_model_priors(pipeline::AbstractEpiAwarePipeline)
1719
transformed_process_init_prior = Normal(0.0, 0.25)
18-
std_prior = HalfNormal(0.25)
19-
damp_param_prior = Beta(0.5, 0.5)
20-
log_I0_prior = Normal(log(100.0), 1e-5)
20+
std_prior = HalfNormal(0.025)
21+
damp_param_prior = Beta(1, 9)
22+
log_I0_prior = Normal(log(100.0), 1e-1)
2123

2224
return Dict(
2325
"transformed_process_init_prior" => transformed_process_init_prior,

0 commit comments

Comments
 (0)