tests/suite/models/01_sleepstudy.R

In jtimonen/lgpr: Longitudinal Gaussian Process Regression

# A model for the sleepstudy data
library(lgpr)
library(lme4)

# Load data
setup_data <- function() {
  lme4::sleepstudy
}

# Create model
setup_model <- function(...) {
  dat <- setup_data()
  model <- create_model(Reaction ~ Days + Days | Subject, dat, ...)
  return(model)
}

expected_relevances <- function() {
  c(0.3096018, 0.5637252, 0.1266730)
}