grf (version 1.1.0)

plot.grf_tree: Plot a GRF tree object.

Description

The direction NAs are sent are indicated with the arrow fill. An empty arrow indicates that NAs are sent that way. If trained without missing values, both arrows are filled.

Usage

# S3 method for grf_tree
plot(x, include.na.path = NULL, ...)

Arguments

x

The tree to plot

include.na.path

A boolean toggling whether to include the path of missing values or not. It defaults to whether the forest was trained with NAs.

...

Additional arguments (currently ignored).

Examples

# NOT RUN {
# Save the plot of a tree in the causal forest.
install.packages("DiagrammeR")
install.packages("DiagrammeRsvg")
n <- 500
p <- 10
X <- matrix(rnorm(n * p), n, p)
W <- rbinom(n, 1, 0.5)
Y <- pmax(X[, 1], 0) * W + X[, 2] + pmin(X[, 3], 0) + rnorm(n)
c.forest <- causal_forest(X, Y, W)
#save the first tree in the forest as plot.svg
tree.plot = plot(get_tree(c.forest, 1))
cat(DiagrammeRsvg::export_svg(tree.plot), file='plot.svg')
# }