R/treesize.R

Defines functions treesize

Documented in treesize

treesize <- function(x, terminal=TRUE) {
  if(!inherits(x, "randomForest"))
    stop("This function only works for objects of class `randomForest'")
  if(is.null(x$forest)) stop("The object must contain the forest component")
  if(terminal) return((x$forest$ndbigtree+1)/2) else return(x$forest$ndbigtree)
}
karlkumbier/irfRegression documentation built on May 18, 2017, 3:28 p.m.