tests/testthat/test-controls.R

In MachineShop: Machine Learning Models and Tools

## Resampling Controls

test_that("formula resampling", {
  skip_if_not(TEST_ALL)
  with_parallel({

    library(MASS)
    library(recipes)

    df1 <- ICHomes
    fo1 <- sale_amount ~ .
    rec1 <- recipe(fo1, data = df1)

    df2 <- iris
    fo2 <- Species ~ .
    rec2 <- recipe(fo2, data = df2)

    df3 <- within(Melanoma, status <- status != 2)
    fo3 <- Surv(time, status) ~ .
    rec3 <- recipe(time + status ~ ., data = df3) %>%
      role_surv(time = time, event = status) %>%
      role_case(stratum = status)

    model <- GBMModel

    controls <- c(
      "BootControl",
      "BootOptimismControl",
      "CVControl",
      "CVOptimismControl",
      "OOBControl",
      "SplitControl",
      "TrainControl"
    )

    for (control in controls) {
      ## formula resampling
      expect_s4_class(resample(fo1, df1, model, control = control), "Resample")
      expect_s4_class(resample(fo2, df2, model, control = control), "Resample")
      expect_s4_class(resample(fo3, df3, model, control = control), "Resample")
      ## recipe resampling
      expect_s4_class(resample(rec1, model, control = control), "Resample")
      expect_s4_class(resample(rec2, model, control = control), "Resample")
      expect_s4_class(resample(rec3, model, control = control), "Resample")
    }

  })
})