tests/testthat/test_engine_lightgbm.R

In five-dots/ml4e: Machine Learning for Everyone

test_that("lightgbm_es for multiclass classification works", {

  keys_cv["model"] <- "lightgbm_es"
  cv <- test_model(options_mc, keys_cv, "lgb.Booster", pred_mc_col_names)

})

test_that("lightgbm_es for binary classification works", {

  keys_cv["model"] <- "lightgbm_es"
  cv <- test_model(options_bin, keys_cv, "lgb.Booster", pred_bin_col_names)

})

test_that("lightgbm_es for regression works", {

  keys_cv["model"] <- "lightgbm_es"
  cv <- test_model(options_reg, keys_cv, "lgb.Booster", pred_reg_col_names)

})

test_that("lightgbm_es for poisson works", {

  keys_cv["model"] <- "lightgbm_es"
  cv <- test_model(options_pois, keys_cv, "lgb.Booster", pred_reg_col_names)

})