tests/testthat/test-ova.R

In AlineTalhouk/splendid: SuPervised Learning ENsemble for Diagnostic IDentification

data(hgsc)
dat <- hgsc[1:100, 1:25]
class <- attr(hgsc, "class.true")[1:100]

test_that("knn ova works", {
  sl_result <- splendid_model(dat, class, n = 1, algorithms = "knn",
                              ova = TRUE)
  expect_is(sl_result$preds$ova_knn[[1]] %@% "prob", "matrix")
})

test_that("mlr_nnet and nnet ova outputs need matrix coercion", {
  sl_result <- splendid_model(dat, class, n = 1,
                              algorithms = c("mlr_nnet", "nnet"),
                              ova = TRUE)
  expect_is(sl_result$preds$ova_mlr_nnet[[1]] %@% "prob", "matrix")
  expect_is(sl_result$preds$ova_nnet[[1]] %@% "prob", "matrix")
})