combineDummy: Update Dummy Variable Names

In AlanInglis/BartVis: Create Visualisations for BART Models

Update Dummy Variable Names

Description

This function updates the 'var' column in the 'structure' component of the 'trees' list, replacing dummy variable names derived from factor variables with their original factor variable names.

Usage

combineDummy(trees)

Arguments

trees

A list containing at least two components: 'data' and 'structure'. 'data' should be a dataframe, and 'structure' a dataframe that includes a 'var' column.

Details

The function first identifies factor variables in 'trees$data', then checks each entry in 'trees$structure$var' for matches with these factor variables. If a match is found, indicating a dummy variable, the entry is replaced with the original factor variable name.

Value

The modified 'trees' list with updated 'var' column entries in 'trees$structure'.

Examples

if(requireNamespace("dbarts", quietly = TRUE)){
# Load the dbarts package to access the bart function
 library(dbarts)
 # Create Simple dbarts Model with Dummies
 set.seed(1701)
 dbartModel <- bart(iris[2:5], iris[,1], ntree = 5, keeptrees = TRUE, nskip = 10, ndpost = 10)
 # Tree Data
 trees_data <- extractTreeData(model = dbartModel, data = iris)
 combined_trees <- combineDummy(trees = trees_data)
}

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