tests/testthat/test_classif_naiveBayes.R

In guillermozbta/s2: Machine Learning in R

context("classif_naiveBayes")

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)
})