add se in extra slot and keep prediction

hiredd · May 12, 2017 · d7c111c · d7c111c
1 parent 91b2ef4
commit d7c111c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/R/predict.R b/R/predict.R
@@ -381,7 +381,10 @@ predict.ranger.forest <- function(object, data, predict.all = FALSE,
     jack <- (n - 1) / n * rowSums((jack.n - yhat)^2)
     bias <- (exp(1) - 1) * n / result$num.trees^2 * rowSums((result$predictions - yhat)^2)
     jab <- pmax(jack - bias, 0)
-    result$predictions <- sqrt(jab)
+    result$se <- sqrt(jab)
+
+    ## Response as predictions
+    result$predictions <- yhat
   }
 
   class(result) <- "ranger.prediction"