File: test_base_costs.R

package info (click to toggle)
r-cran-mlr 2.19.1%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 8,392 kB
  • sloc: ansic: 65; sh: 13; makefile: 5
file content (21 lines) | stat: -rwxr-xr-x 678 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

test_that("costs", {
  lrn = makeLearner("classif.rpart")
  rdesc = makeResampleDesc("Holdout")
  task = binaryclass.task
  task$task.desc$positive = "M"

  cc = 1 - diag(1, 2)
  rownames(cc) = colnames(cc) = getTaskClassLevels(task)
  ms = makeCostMeasure(costs = cc)
  r = resample(lrn, rdesc, task = task, measures = list(mmce, ms))
  expect_equal(r$aggr[[1]], r$aggr[[2]])

  cc = matrix(0, 2, 2)
  rownames(cc) = colnames(cc) = getTaskClassLevels(task)
  cc["R", "M"] = 1
  ms = makeCostMeasure(id = "foo", costs = cc, combine = sum)
  expect_equal(ms$id, "foo")
  r = resample(lrn, rdesc, task = task, measures = list(fp, ms))
  expect_equal(r$aggr[[1]], r$aggr[[2]])
})