RJAGS simulation with categorical variables

Consider the Normal regression model of volume \(Y\)_i by weekday status \(X\)_i:

likelihood: \(Y\)_i \(\sim N(m\)_i, \(s^2)\) where \(m\)_i \(= a + b X\)_i
priors: \(a \sim N(400, 100^2)\), \(b \sim N(0, 200^2)\), \(s \sim Unif(0, 200)\)

You explored the relationship between \(Y\)_i and \(X\)_i for the 90 days recorded in RailTrail (in your workspace). In light of these data and the priors above, you will update your posterior model of this relationship. This differs from previous analyses in that \(X\)_i is categorical. In rjags syntax, its coefficient \(b\) is defined by two elements, b[1] and b[2], which correspond to the weekend and weekday levels, respectively. For reference, b[1] is set to 0. In contrast, b[2] is modeled by the prior for \(b\).

Cet exercice fait partie du cours

Bayesian Modeling with RJAGS

Afficher le cours

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# DEFINE the model    
rail_model_1 <- "model{
    # Likelihood model for Y[i]
    for(i in ___){
      Y[i] ~ ___
      m[i] <- ___
    }
    
    # Prior models for a, b, s
    a ~ ___
    b[1] <- ___
    b[2] ~ ___
    s ~ ___
}"

Modifier et exécuter le code