diff --git a/ranger-r-package/ranger/R/predict.R b/ranger-r-package/ranger/R/predict.R index e5cecfced..89bd6d5fa 100644 --- a/ranger-r-package/ranger/R/predict.R +++ b/ranger-r-package/ranger/R/predict.R @@ -225,6 +225,9 @@ predict.ranger.forest <- function(object, data, predict.all = FALSE, } ##' Prediction with new data and a saved forest from Ranger. +##' +##' For classification and predict.all = TRUE, a matrix of factor levels is returned. +##' To retrieve the corresponding factor levels, use rf$forest$levels, if rf is the ranger object. ##' ##' @title Ranger prediction ##' @param object Ranger \code{ranger} object. diff --git a/ranger-r-package/ranger/man/predict.ranger.Rd b/ranger-r-package/ranger/man/predict.ranger.Rd index 5734d9a85..d1052449f 100644 --- a/ranger-r-package/ranger/man/predict.ranger.Rd +++ b/ranger-r-package/ranger/man/predict.ranger.Rd @@ -38,6 +38,10 @@ Object of class \code{ranger.prediction} with elements \description{ Prediction with new data and a saved forest from Ranger. } +\details{ +For classification and predict.all = TRUE, a matrix of factor levels is returned. +To retrieve the corresponding factor levels, use rf$forest$levels, if rf is the ranger object. +} \author{ Marvin N. Wright }