2024-12-16 update : fixing params funx & docs.

cbernalz · Dec 16, 2024 · a50e7de · a50e7de
1 parent fe48736
commit a50e7de
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/docs/src/tutorials/uciwwiehr_model_fitting_forecast.md b/docs/src/tutorials/uciwwiehr_model_fitting_forecast.md
@@ -17,12 +17,12 @@ Here we generate two datasets, one with 150 time points and one with 178 time po
 using UCIWWEIHR
 # Running simulation function with presets
 params = create_uciwweihr_sim_params(
-    time_points = 150
+    time_points = 70
 )
 df = generate_simulation_data_uciwweihr(params)
 
 params_ext = create_uciwweihr_sim_params(
-    time_points = 178
+    time_points = 84
 )
 df_ext = generate_simulation_data_uciwweihr(params_ext)
 first(df, 5)
@@ -46,7 +46,7 @@ param_change_times = 1:7:max_obstime # Change every week
 priors_only = false
 n_samples = 200
 forecast = true
-forecast_weeks = 4
+forecast_weeks = 2
 
 model_params = create_uciwweihr_model_params()
 samples = uciwweihr_fit(

diff --git a/src/uciwweihr_model_params.jl b/src/uciwweihr_model_params.jl
@@ -79,21 +79,21 @@ Creates a `uciwweihr_sim_params` struct with the option to either use a predeter
 - `params::uciwweihr_sim_params`: A struct with simulation parameters.
 """
 function create_uciwweihr_model_params(; 
-    E_init_sd::Float64=50.0, log_E_init_mean::Float64=log(200),
-    I_init_sd::Float64=20.0, log_I_init_mean::Float64=log(100),
-    H_init_sd::Float64=5.0, log_H_init_mean::Float64=log(20),
+    E_init_sd::Float64=0.2, log_E_init_mean::Float64=log(200),
+    I_init_sd::Float64=0.2, log_I_init_mean::Float64=log(100),
+    H_init_sd::Float64=0.2, log_H_init_mean::Float64=log(20),
     gamma_sd::Float64=0.02, log_gamma_mean::Float64=log(1/4),
     nu_sd::Float64=0.02, log_nu_mean::Float64=log(1/7),
     epsilon_sd::Float64=0.02, log_epsilon_mean::Float64=log(1/5),
     rho_gene_sd::Float64=0.02, log_rho_gene_mean::Float64=log(0.011),
 
-    sigma_wastewater::Union{Float64, Nothing}=0.1,
-    sigma_hosp::Union{Float64, Nothing}=500.0,
+    sigma_wastewater::Union{Float64, Nothing}=nothing,
+    sigma_hosp::Union{Float64, Nothing}=nothing,
 
-    #sigma_ww_sd::Float64=0.02, log_sigma_ww_mean::Float64=log(0.1),
-    #sigma_hosp_sd::Float64=50.0, sigma_hosp_mean::Float64=500.0,
-    sigma_ww_sd::Union{Float64, Nothing}=nothing, log_sigma_ww_mean::Union{Float64, Nothing}=nothing,
-    sigma_hosp_sd::Union{Float64, Nothing}=nothing, sigma_hosp_mean::Union{Float64, Nothing}=nothing,
+    sigma_ww_sd::Float64=0.02, log_sigma_ww_mean::Float64=log(0.1),
+    sigma_hosp_sd::Float64=50.0, sigma_hosp_mean::Float64=500.0,
+    #sigma_ww_sd::Union{Float64, Nothing}=nothing, log_sigma_ww_mean::Union{Float64, Nothing}=nothing,
+    #sigma_hosp_sd::Union{Float64, Nothing}=nothing, sigma_hosp_mean::Union{Float64, Nothing}=nothing,
 
 
     Rt_init_sd::Float64=0.3, Rt_init_mean::Float64=0.2,