tests/testthat/test_classif_rFerns.R

In mlr-org/mlr: Machine Learning in R

test_that("classif_rFerns", {
  requirePackagesOrSkip("rFerns", default.method = "load")

  parset.list = list(
    list(),
    list(ferns = 200L, depth = 4L)
  )

  old.predicts.list = list()

  for (i in seq_along(parset.list)) {
    parset = parset.list[[i]]
    parset = c(list(formula = binaryclass.formula, data = binaryclass.train),
      parset)
    set.seed(getOption("mlr.debug.seed"))
    m = do.call(rFerns::rFerns, parset)
    old.predicts.list[[i]] = factor(predict(m, binaryclass.test))
  }

  testSimpleParsets("classif.rFerns", binaryclass.df, binaryclass.target,
    binaryclass.train.inds,
    old.predicts.list, parset.list)
})