tests/testthat/test_classif_naiveBayes.R

In berndbischl/mlr: Machine Learning in R

test_that("classif_naiveBayes", {
  requirePackagesOrSkip("e1071", default.method = "load")

  m = e1071::naiveBayes(formula = multiclass.formula, data = multiclass.train)
  p = predict(m, newdata = multiclass.test[, -multiclass.class.col])
  p2 = predict(m, newdata = multiclass.test[, -multiclass.class.col],
    type = "raw")

  testSimple("classif.naiveBayes", multiclass.df, multiclass.target,
    multiclass.train.inds, p)
  testProb("classif.naiveBayes", multiclass.df, multiclass.target,
    multiclass.train.inds, p2)

  tt = e1071::naiveBayes
  tp = function(model, newdata) predict(model, newdata[, -multiclass.class.col])

  testCV("classif.naiveBayes", multiclass.df, multiclass.target,
    tune.train = tt, tune.predict = tp)
})