caret multiclass

gtesei · Mar 23, 2016 · 97963a7 · 97963a7
1 parent 7b24cdf
commit 97963a7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/R-package/R/fastRegression.R b/R-package/R/fastRegression.R
@@ -1197,9 +1197,12 @@ ff.createEnsemble = function(Xtrain,
 }
 
 getCaretFactors = function(y) {
-  stopifnot(sort(unique(y))[1] == 0, sort(unique(y))[2] == 1, length(unique(y)) == 2)
+  ncl = length(unique(unique(y)))
+  for (i in 1:ncl) {
+    stopifnot(sort(unique(y))[i] == (i-1))
+  }
   y.cat = factor(y) 
-  levels(y.cat) = c("class0","class1")
+  levels(y.cat) = paste0("class",0:(ncl-1))
   return(list(y.cat=y.cat,fact.sign="class1"))
 }
 

diff --git a/fastfurious-manual.pdf b/fastfurious-manual.pdf