tests/testthat/test-plot_roc.SuperLearner.R

In ck37/ckTools: Chris Kennedy's R Toolkit

library(SuperLearner)
library(ck37r)

# Only run test if necessary suggested packages are installed.
pkg_suggests = c("testthat", "MASS")
if (!all(suppressWarnings(sapply(pkg_suggests, require, quietly = TRUE,
                                 character.only = TRUE))))
  return()

data(Boston, package = "MASS")

set.seed(1)
y = as.integer(Boston$age > 68)
sl = SuperLearner(y,
                  subset(Boston, select = -age),
                  family = binomial(),
                  cvControl = list(V = 3L, stratifyCV = TRUE),
                  SL.library = c("SL.mean", "SL.glm"))

sl

plot_roc(sl, y = y)

# Deprecated version, will generate a warning.
suppressWarnings({
  sl_plot_roc(sl, y = y)
})