tests/testthat/test_resample.R

In mlr-org/mlr3ordinal: Ordinal regression for mlr3

context("resampling")

test_that("resampling works for ordinal learners", {
  rr = resample(tsk("winerating"), lrn("ordinal.clm"), rsmp("cv"))

  task = tsk("winerating")
  task2 = task$clone()
  task2$col_roles$stratum = task2$target_names

  expect_resampling(rr$resampling, task = rr$task)
  expect_task(rr$task)
  expect_resample_result(rr)
})

test_that("resampling works for regression learners", {
  learner = lrn("regr.rpart")
  graph = ppl("ordinal", learner)
  glearner = GraphLearner$new(graph, task_type = "ordinal")
  rr = resample(tsk("winerating"), glearner, rsmp("cv"))
  expect_resample_result(rr)
})

test_that("resampling works for classification learners", {
  learner = lrn("classif.rpart")
  graph = ppl("ordinal", learner)
  glearner = GraphLearner$new(graph, task_type = "ordinal")
  rr = resample(tsk("winerating"), glearner, rsmp("cv"))
  expect_resample_result(rr)
})