vintPlot: vintPlot

In AlanInglis/BartVis: Create Visualisations for BART Models

vintPlot

Description

Plot the pair-wise variable interactions inclusion porportions for a BART model with the 25

Usage

vintPlot(trees, plotType = "barplot", top = NULL)

Arguments

trees	A data frame created by 'extractTreeData' function.
plotType	Which type of plot to return. Either a barplot 'barplot' with the quantiles shown as a line, a point plot with the quantiles shown as a gradient 'point', or a letter-value plot 'lvp'.
top	Display only the top X metrics (does not apply to the letter-value plot).

Value

A plot of variable importance.

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)
 vintPlot(trees = trees_data, top = 5)
}

AlanInglis/BartVis documentation built on July 27, 2024, 12:02 a.m.