tests/testthat/test_mlr_measures_classif.auc.R

test_that("mlr_measures_auc", {
  task = tsk("sonar")
  lrn = lrn("classif.featureless", predict_type = "prob")
  m = msr("classif.auc")
  measures = list(m)

  p = lrn$train(task)$predict(task)
  perf = p$score(measures)
  expect_equal(unname(perf), 0.5)
})

Try the mlr3 package in your browser

Any scripts or data that you put into this service are public.

mlr3 documentation built on Nov. 17, 2023, 5:07 p.m.