revdep/checks.noindex/MachineShop/old/MachineShop.Rcheck/tests/testthat/test-ML_SuperModel.R

context("SuperModel")

test_that("model fitting", {
  skip_if_not(TEST_MODEL_FITTING)
  with_parallel({
    model <- SuperModel(GBMModel, SVMRadialModel, KNNModel)
    expect_output(test_model_binary(model))
    expect_output(test_model_factor(model))
    expect_output(test_model_numeric(model))

    model <- SuperModel(RandomForestModel, POLRModel)
    expect_output(test_model_ordered(model))

    model <- SuperModel(GBMModel, GLMNetModel(lambda = 0.01))
    expect_output(test_model_Surv(model))
  })
})
gbm-developers/gbm documentation built on Feb. 16, 2024, 6:13 p.m.