fixed object name bug

topepo · Jul 15, 2015 · 5f1e416 · 5f1e416
1 parent aed5b84
commit 5f1e416
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/models/files/bartMachine.R b/models/files/bartMachine.R
@@ -55,12 +55,12 @@ modelInfo <- list(label = "Bayesian Additive Regression Trees",
                     },
                   predictors = function(x, ...)  colnames(x$X),
                   varImp = function(object, ...){
-                    imps <- investigate_var_importance(x, plot = FALSE)
+                    imps <- investigate_var_importance(object, plot = FALSE)
                     imps <- imps$avg_var_props - 1.96*imps$sd_var_props
-                    missing_x <- !(colnames(x$X) %in% names(imps))
+                    missing_x <- !(colnames(object$X) %in% names(imps))
                     if(any(missing_x)) {
                       imps2 <- rep(0, sum(missing_x))
-                      names(imps2) <- colnames(x$X)[missing_x]
+                      names(imps2) <- colnames(object$X)[missing_x]
                       imps <- c(imps, imps2)
                     }
                     out <- data.frame(Overall = as.vector(imps))