tests/testthat/test_ensemble_voting.R

In orionw/BestModel: A package to run and compare standard machine learning models with easy visualization

context("Voting with Ensemble Functionality")

# load the libraries
library(ModelComparison)

test_that("Ensemble weighting function with auto-weighting", {
  # prepare the dataset
  titanic <- PrepareNumericTitanic()
  # create the models
  comp <- GetModelComparisons(titanic[, -1], titanic[, 1])
  ensem <- Ensemble(comp$model.list, "majorityWeight", titanic[, -1], titanic[, 1])
  expect_equal(length(ensem$weight.list), length(ensem$models))
  expect_equal(class(ensem), "Ensemble")
  pred <- predict(ensem, titanic[, -1])
  expect_equal(length(titanic[, 1]), length(pred))
})

test_that("Ensemble weighting function with given weights", {
  # prepare the dataset
  titanic <- PrepareNumericTitanic()
  # create the models
  comp <- GetModelComparisons(titanic[, -1], titanic[, 1])
  ensem <- Ensemble(comp$model.list, "majorityWeight", runif(n=length(comp$model.list), min=0, max=1))
  expect_equal(class(ensem), "Ensemble")
  pred <- predict(ensem, titanic[, -1])
  expect_equal(length(titanic[, 1]), length(pred))
})

test_that("Average weighting Ensemble", {
  # prepare the dataset
  titanic <- PrepareNumericTitanic()
  # create the models
  comp <- GetModelComparisons(titanic[, -1], titanic[, 1])
  ensem <- Ensemble(comp$model.list, "averageVote", runif(n=length(comp$model.list), min=0, max=1))
  expect_equal(class(ensem), "Ensemble")
  pred <- predict(ensem, titanic[, -1])
  expect_equal(length(titanic[, 1]), length(pred))
})

test_that("Majority Vote Ensemble", {
  # prepare the dataset
  titanic <- PrepareNumericTitanic()
  # create the models
  comp <- GetModelComparisons(titanic[, -1], titanic[, 1])
  ensem <- Ensemble(comp$model.list, "majorityVote", runif(n=length(comp$model.list), min=0, max=1))
  expect_equal(class(ensem), "Ensemble")
  pred <- predict(ensem, titanic[, -1])
  expect_equal(length(titanic[, 1]), length(pred))
})

test_that("Majority Vote Ensemble With only 2 models", {
  # prepare the dataset
  titanic <- PrepareNumericTitanic()
  # create the models
  comp <- GetModelComparisons(titanic[, -1], titanic[, 1], model.list = "all")
  models <- comp$model.list[c(F, T, T, F, F, F)]
  ensem <- Ensemble(models, "majorityVote")
  expect_equal(class(ensem), "Ensemble")
  pred <- predict(ensem, titanic[, -1])
  expect_equal(length(titanic[, 1]), length(pred))
})