Black boost multinomial restrictions for issue #779

topepo · Nov 27, 2017 · 1c1eff0 · 1c1eff0
1 parent 3f60e63
commit 1c1eff0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/models/files/blackboost.R b/models/files/blackboost.R
@@ -27,6 +27,10 @@ modelInfo <- list(label = "Boosted Tree",
                   fit = function(x, y, wts, param, lev, last, classProbs, ...) { 
                     theDots <- list(...)
 
+                    if(length(levels(y)) > 2) 
+                      stop("Two-class outcomes only. See ?mboost::Multinomial",
+                           call. = FALSE)
+
                     if(any(names(theDots) == "tree_controls")) {
                       theDots$tree_controls$maxdepth <- param$maxdepth 
                       treeCtl <- theDots$tree_controls

diff --git a/pkg/caret/inst/models/models.RData b/pkg/caret/inst/models/models.RData