plotSingleTree {bartMan}R Documentation

plotSingleTree

Description

Plots individual trees.

Usage

plotSingleTree(trees, iter = 1, treeNo = 1, plotType = "icicle")

Arguments

trees

A data frame created by extractTreeData function

iter

The MCMC iteration or chain to plot.

treeNo

The tree number to plot.

plotType

What type of plot to display. either dendrogram or icicle.

Value

A plot of an individual tree

Examples

if (requireNamespace("dbarts", quietly = TRUE)) {
 # Load the dbarts package to access the bart function
 library(dbarts)
 # Get Data
 df <- na.omit(airquality)
 # Create Simple dbarts Model For Regression:
 set.seed(1701)
 dbartModel <- bart(df[2:6],
   df[, 1],
   ntree = 5,
   keeptrees = TRUE,
   nskip = 10,
   ndpost = 10
 )
 # Tree Data
 trees_data <- extractTreeData(model = dbartModel, data = df)
 plotSingleTree(trees = trees_data, iter = 1, treeNo = 1)
}

[Package bartMan version 0.1.1 Index]