viviBart {bartMan}

viviBart

Description

Returns a list containing a dataframe of variable importance summaries and a dataframe of variable interaction summaries.

Usage

viviBart(trees, out = "vivi")

Arguments

trees

A data frame created by 'extractTreeData' function.

out

Choose to either output just the variable importance ('vimp'), the variable interaction ('vint'), or both ('vivi') (default).

Value

A list of dataframes of VIVI summaries.

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)
 viviBart(trees = trees_data, out = 'vivi')
 }