bartMan (version 0.1.1)

combineDummy: 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)

Value

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

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.

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)
}