Let’s load the necessary packages:
library(zoid)
We will use the “broken stick” approach to simulate data from the Dirichlet - trinomial model. This model assumes that the group proportions for each observation are Dirichlet, but the observed values are either 0, the total sample size (N) or a number between 0 and N.
Our broken_stick
function can be called as follows,
= broken_stick(n_obs = 10,
y n_groups = 10,
tot_n = 100)
The object y
is a list with 2 elements, (1) the true underlying compositions (p) and the realized data (X_obs). They can be accessed as
$p
y$X_obs y
By default, the simulation function assumes a uniform prior for the Dirichlet, with hyperparameters = 1. We can change this by specifying our own values of hyperparameters. Using the argument p
, we can simulate new values with a slightly larger effective sample size, and pass that into broken_stick
= gtools::rdirichlet(1, alpha = rep(2,10))
p
= broken_stick(n_obs = 10,
y n_groups = 10,
tot_n = 100,
p = p)