tests/testthat/test_learner_classif_lightgbm.R

context("LearnerClassifLightGBM")

test_that(
  desc = "LearnerClassifLightGBM",
  code = {
    learner = LearnerClassifLightGBM$new()
    expect_learner(learner)
    learner$param_set$values = mlr3misc::insert_named(
      learner$param_set$values,
      list(
        "early_stopping_round" = 3,
        "learning_rate" = 0.1,
        "num_iterations" = 10
      )
    )
    result = run_autotest(learner)
    expect_true(result, info = result$error)
  }
)
kapsner/mlr3learners.lightgbm documentation built on Feb. 17, 2021, 5:53 p.m.