tests/testthat/test_MeasureRegr.R

In mlr3: Machine Learning in R - Next Generation

test_that("Regression measures", {
  keys = mlr_measures$keys("^regr\\.")
  task = tsk("boston_housing")
  learner = lrn("regr.rpart")
  learner$train(task)
  p = learner$predict(task)

  for (key in keys) {
    m = mlr_measures$get(key)
    if (is.na(m$task_type) || m$task_type == "regr") {
      perf = m$score(prediction = p, task = task, learner = learner)
      expect_number(perf, na.ok = FALSE, lower = m$range[1], upper = m$range[2])
    }
  }
})