tests/testthat/test_base_getOOBPreds.R

In Najah-lshanableh/R-data-mining2: Machine Learning in R

context("getOOBPreds")

test_that("getOOBPreds", {
  lrn = makeLearner("classif.randomForest")
  task = subsetTask(binaryclass.task, subset = c(10:20, 180:190),
    features = getTaskFeatureNames(binaryclass.task)[12:15])
  mod = train(lrn, task)
  oob = getOOBPreds(mod, task)
  pred = predict(mod, task)
  expect_true(is.numeric(performance(oob, measures = list(acc))))
  expect_equal(dim(oob$data), dim(pred$data))
  expect_equal(names(oob$data), names(pred$data))
  expect_equal(names(oob), names(pred))
})