# Custom distributions for uncertainty propagation

    library(tidyverse)
    library(biogrowth)

C:6f2837a024705_custom_stochastic.R

The biogrowth package includes two functions for making simulations including parameter uncertainty: `predict_growth_uncertainty()` and the generic `predictMCMC()`. The former includes parameter uncertainty considering that the model parameters follow a normal distribution with known mean and variance (defined on different scales). The latter, uses the Markov Chain of a model fitted using an MCMC algorithm to propagate the uncertainty in the parameter estimates.

Although these functions account for parameter uncertainty in the model predictions, they do not allow the definition of custom distribution for the model parameters. Nonetheless, that is relatively simple to do using the `predict_growth()` function together with the functions from the tidyverse.

First, we need to define the growth model to use and the time points where the solution is to be calculated. This is done in the same way as a “usual” calculation in `predict_growth()`:

    my_model <- "modGompertz"
    my_time <- seq(0, 100, length = 1000)

C:6f2837a024705_custom_stochastic.R

Next, we need to define the parameter sample. The `tibble()` function provides a convenient way for this. In this example, we will use 500 iterations, considering that \(\log N \sim Normal(0,1)\), \(\mu \sim Gamma(3,5)\) and \(\lambda \sim Gamma(2,2)\) with \(C=6\) constant.

    set.seed(12412)
    niter <- 500

    par_sample <- tibble(logN0 = rnorm(niter, mean = 0, sd = 1),
                         C = 6,
                         mu = rgamma(niter, shape = 3, rate = 5),
                         lambda = rgamma(niter, shape = 2, rate = 2))

    par_sample %>%
        pivot_longer(everything()) %>%
        ggplot() + geom_histogram(aes(value)) + facet_wrap("name", scales = "free")
    #> `stat_bin()` using `bins = 30`. Pick better value with `binwidth`.

C:6f2837a024705_custom_stochastic.R

The `pmap()` function from purrr provides a convenient way to convert the parameter sample to a list that defines the model and we can pass to `predict_growth()` using the `map()` function (also from purrr).

    my_predictions <- par_sample %>%
        pmap(., function(logN0, mu, lambda, C)
            list(model = my_model,
                 logN0 = logN0,
                 mu = mu,
                 lambda = lambda,
                 C = C)
            ) %>%
        map(.,
            ~ predict_growth(my_time, .)
            )

C:6f2837a024705_custom_stochastic.R

Now, it is just a question of post-processing the simulations. For instance, we can calculate summary statistics of the simulations at each time point.

    summary_preds <- my_predictions %>%
        map(., ~.$simulation) %>%
        imap_dfr(., ~ mutate(.x, sim = .y)) %>%
        group_by(time) %>%
        summarize(m_logN = median(logN),
                  q10 = quantile(logN, .1),
                  q90 = quantile(logN, .9))

C:6f2837a024705_custom_stochastic.R

…that can be plotted using `ggplot()`

    summary_preds %>%
        ggplot(aes(x = time)) +
        geom_ribbon(aes(ymin = q10, ymax = q90), alpha = .5) +
        geom_line(aes(y = m_logN))

C:6f2837a024705_custom_stochastic.R