diff --git a/docs/src/tutorials/uciwwiehr_model_fitting_forecast.md b/docs/src/tutorials/uciwwiehr_model_fitting_forecast.md index c5d12db..2434403 100644 --- 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 index fcc57e5..7722c71 100644 --- 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,