tests/testthat/test_basic.R

In deepboost: Deep Boosting Ensemble Modeling

library(deepboost)

context("basic functions")

data(adult, package='deepboost')

set.seed(666)

formula <- X..50K ~ X39 + X77516 + X13 + X2174 +  X0 + X40
levels(adult[,length(adult)]) <- c(1,-1)

train <- adult[1:29000,]
test <- adult[29001:32560,]

test_that("train and predict formula works", {
  bst <- deepboost.formula(formula, train, num_iter = 5)
  pred <- predict(bst, test)
  expect_equal(length(pred), 3560)
})

test_that("train and predict default works", {
  bst <- deepboost.default(train[,c("X39","X77516","X13")], train$X..50K, num_iter = 5)
  pred <- predict(bst, test)
  expect_equal(length(pred), 3560)
})

test_that("train and predict probs works", {
  bst <- deepboost.default(train[,c("X39","X77516","X13")], train$X..50K, num_iter = 5)
  pred <- predict(bst, test, type="response")
  expect_equal(nrow(pred), 3560)
})

test_that("predict labels and predict probs output similar decisions", {
  bst <- deepboost.default(train[,c("X39","X77516","X13")], train$X..50K, num_iter = 5)
  predLabels <- predict(bst, test)
  predProbabilities <- predict(bst, test, type="response")
  expect_equal((predLabels=="1"), (predProbabilities[,1] > 0.5))
})

test_that("grid search works", {
  best_params <-
    deepboost.gridSearch(x1 ~ x2, data.frame(x1=rep(c(1,1,1,0),5),x2=rep(c(1,1,1,1),5)),
                         seed = 666, k = 2)

  expect_equal(is.numeric(best_params[1][[1]]), TRUE)
  expect_equal(is.numeric(best_params[2][[1]]), TRUE)
  expect_equal(is.numeric(best_params[3][[1]]), TRUE)
  expect_equal(is.numeric(best_params[4][[1]]), TRUE)
  expect_equal(is.character(best_params[5][[1]]), TRUE)

})