tests/testthat/test-get_coef.R

In sgboost: Sparse-Group Boosting

test_that("get_coef works", {
  expect_error(
    get_varimp(sgb_model = "sgb"),
    "Model must be of class mboost"
  )
  library(dplyr)
  library(mboost)
  set.seed(1)
  df <- data.frame(
    x1 = rnorm(100), x2 = rnorm(100), x3 = rnorm(100),
    x4 = rnorm(100), x5 = runif(100)
  )
  df <- df %>%
    mutate_all(function(x) {
      as.numeric(scale(x))
    })
  df$y <- df$x1 + df$x4 + df$x5
  group_df <- data.frame(
    group_name = c(1, 1, 1, 2, 2),
    var_name = c("x1", "x2", "x3", "x4", "x5")
  )
  sgb_formula <- as.formula(create_formula(alpha = 0.3, group_df = group_df, intercept = T))
  sgb_model <- mboost(formula = sgb_formula, data = df)
  sgb_coefs <- get_coef(sgb_model = sgb_model)[[2]] %>% select(variable, effect)
  expect_equal(
    tibble(
      variable = c("(Intercept)", "x3", "x2", "x1", "x5", "x4"),
      effect = c(0.000, -0.006, -0.021, 0.226, 0.508, 0.569)
    ) %>%
      arrange(-abs(effect)),
    sgb_coefs %>%
      mutate(effect = round(effect, 3))
  )
})