R/meanvar.causalTree.R

In swager/causalForest: Causal Trees and Forests

meanvar.causalTree <- function(tree, xlab = "ave(y)", ylab = "ave(deviance)", ...)

{
    if (!inherits(tree, "causalTree"))
        stop("Not a legitimate \"causalTree\" object")
    if (!tree$method == "anova")
        stop("Plot not useful for classification or poisson trees")
    frame <- tree$frame
    frame <- frame[frame$var == "<leaf>", ]
    x <- frame$yval
    y <- frame$dev/frame$n
    label <- row.names(frame)
    plot(x, y, xlab = xlab, ylab = ylab, type = "n", ...)
    text(x, y, label)
    invisible(list(x = x, y = y, label = label))
}

meanvar <- function(tree, ...) UseMethod("meanvar")