Change to predict.train to fix issue #574

topepo · Mar 24, 2017 · 2576c15 · 2576c15
1 parent 7f5d897
commit 2576c15
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/caret/R/predict.train.R b/pkg/caret/R/predict.train.R
@@ -148,6 +148,8 @@ predict.train <- function(object, newdata = NULL, type = "raw", na.action = na.o
       }
     } else stop("please specify data via newdata")
   }
+  if("xNames" %in% names(object$finalModel))
+    newdata <- newdata[, colnames(newdata) %in% object$finalModel$xNames, drop = FALSE]
 
   if(type == "prob") {
     out <- probFunction(method = object$modelInfo,