revdep/checks.noindex/MachineShop/new/MachineShop.Rcheck/tests/testthat/test-TrainedModels.R

In gbm-developers/gbm: Generalized Boosted Regression Models

context("Trained Models")


models <- c(
  "AdaBagModel",
  "AdaBoostModel",
  "BARTMachineModel",
  "BlackBoostModel",
  "C50Model",
  "CForestModel",
  "EarthModel",
  "FDAModel",
  "GAMBoostModel",
  "GBMModel",
  "GLMBoostModel",
  "GLMNetModel",
  "KNNModel",
  "LDAModel",
  "MDAModel",
  "NNetModel",
  "PLSModel",
  "RandomForestModel",
  "RangerModel",
  "RPartModel",
  "SVMANOVAModel",
  "SVMBesselModel",
  "SVMLaplaceModel",
  "SVMLinearModel",
  "SVMPolyModel",
  "SVMRadialModel",
  "SVMTanhModel",
  "XGBDARTModel",
  "XGBLinearModel",
  "XGBTreeModel"
)

control <- CVControl(folds = 3)

test_tune_grid1 <- function(model) {
  fit(type ~ ., data = Pima.tr,
      model = TunedModel(model, grid = 1, control = control))
}

test_tune_grid2 <- function(model) {
  fit(type ~ ., data = Pima.tr,
      model = TunedModel(model, grid = 3, control = control))
}

test_tune_grid3 <- function(model) {
  fit(type ~ ., data = Pima.tr,
      model = TunedModel(model, grid = Grid(length = 100, random = 25),
                         control = control))
}


for (model in models) {
  test_that("grid tuning of models", {
    skip_if_not(TEST_TRAINING)
    context(paste("Tuned Model:", model))
    with_parallel({
      expect_is(test_tune_grid1(model), "MLModelFit")
      expect_is(test_tune_grid2(model), "MLModelFit")
      expect_is(test_tune_grid3(model), "MLModelFit")
    })
  })
}