Skip to contents
library(INLAvaan)

mod <- "
  # intercept with coefficients fixed to 1
  i =~  1*Day0 + 1*Day1 + 1*Day2 + 1*Day3 + 1*Day4 +
        1*Day5 + 1*Day6 + 1*Day7 + 1*Day8 + 1*Day9

  # slope with coefficients fixed to 0:9 (number of days)
  s =~  0*Day0 + 1*Day1 + 2*Day2 + 3*Day3 + 4*Day4 +
        5*Day5 + 6*Day6 + 7*Day7 + 8*Day8 + 9*Day9

  i ~~ i
  i ~ 1

  s ~~ s
  s ~ 1

  i ~~ s

  # fix intercepts
  Day0 ~ 0*1
  Day1 ~ 0*1
  Day2 ~ 0*1
  Day3 ~ 0*1
  Day4 ~ 0*1
  Day5 ~ 0*1
  Day6 ~ 0*1
  Day7 ~ 0*1
  Day8 ~ 0*1
  Day9 ~ 0*1

  # apply equality constraints
  Day0 ~~ v*Day0
  Day1 ~~ v*Day1
  Day2 ~~ v*Day2
  Day3 ~~ v*Day3
  Day4 ~~ v*Day4
  Day5 ~~ v*Day5
  Day6 ~~ v*Day6
  Day7 ~~ v*Day7
  Day8 ~~ v*Day8
  Day9 ~~ v*Day9
  "
dat <- reshape(
  lme4::sleepstudy,
  timevar = "Days",
  idvar = "Subject",
  direction = "wide"
)
names(dat) <- sub("^Reaction\\.(.*)$", "Day\\1", names(dat))
fit <- agrowth(mod, dat)
#> ℹ Finding posterior mode.
#> ✔ Finding posterior mode. [82ms]
#> 
#> ℹ Computing the Hessian.
#> ✔ Computing the Hessian. [73ms]
#> 
#> ℹ Performing VB correction.
#> ✔ VB correction; mean |δ| = 0.104σ. [162ms]
#> 
#> ⠙ Fitting skew normal to 0/6 marginals.
#> ✔ Fitting skew normal to 6/6 marginals. [171ms]
#> 
#> ℹ Sampling covariances and defined parameters.
#> ✔ Sampling covariances and defined parameters. [79ms]
#> 
#> ⠙ Computing ppp and DIC.
#> ✔ Computing ppp and DIC. [379ms]
#> 
coef(fit)
#>      i~~i       i~1      s~~s       s~1      i~~s         v         v         v 
#> 12102.824    32.765    39.182     8.520     0.107   622.400   622.400   622.400 
#>         v         v         v         v         v         v         v 
#>   622.400   622.400   622.400   622.400   622.400   622.400   622.400