treeBarPlot: Plot Frequency of Tree Structures

In AlanInglis/BartVis: Create Visualisations for BART Models

Plot Frequency of Tree Structures

Description

Generates a bar plot showing the frequency of different tree structures represented in a list of tree graphs. Optionally, it can filter to show only the top N trees and handle stump trees specially.

Usage

treeBarPlot(trees, iter = NULL, topTrees = NULL, removeStump = FALSE)

Arguments

trees	A list of tree graphs to display
iter	Optional; specifies the iteration to display.
topTrees	Optional; the number of top tree structures to display. If NULL, displays all.
removeStump	Logical; if TRUE, trees with no edges (stumps) are excluded from the display

Details

This function processes a list of tree structures to compute the frequency of each unique structure, represented by a bar plot. It has options to exclude stump trees (trees with no edges) and to limit the plot to the top N most frequent structures.

Value

A 'ggplot' object representing the bar plot of tree frequencies.

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)
 plot <- treeBarPlot(trees = trees_data, topTrees = 3, removeStump = TRUE)
}

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