tests/testthat/test_CoxBoost_surv_cv_coxboost.R

In mlr-org/mlr3extralearners: Extra Learners For mlr3

test_that("autotest", {
  set.seed(1)
  learner = lrn("surv.cv_coxboost", maxstepno = 1, K = 2)
  expect_learner(learner)
  result = run_autotest(learner, N = 10, check_replicable = FALSE)
  expect_true(result, info = result$error)
})

test_that("optim", {
  set.seed(1)
  learner = lrn("surv.cv_coxboost",
    penalty = "optimCoxBoostPenalty",
    maxstepno = 1, minstepno = 0, iter.max = 1, K = 2)
  task = tsk("rats")
  task$select(c("litter", "rx"))
  expect_prediction_surv(
    suppressWarnings(learner$train(task, row_ids = 1:50)$predict(task, row_ids = 51:100))
  )
  expect_equal(learner$selected_features(at_step = 0), character())
  expect_equal(learner$selected_features(), "litter")
  expect_error(learner$selected_features(at_step = 2)) # maxstepno = 1
})