dtGAP (version 0.0.2)

prepare_tree: Prepare Tree Plot Data for Visualization

Description

This function processes a tree model's output and prepares node and segment data for visualization using ggplot2 or other plotting tools. It supports various tree model formats such as rpart, party, C50, and caret.

Usage

prepare_tree(tree_res, model = c("rpart", "party", "C50", "caret", "cforest"))

Value

A list with two elements:

plot_data

A data frame of node-level information with labels for visualization.

branches

A data frame of edge (branch) coordinates for connecting parent and child nodes.

Arguments

tree_res

A list object containing tree plotting information, including a plot_data data frame.

model

A string indicating the tree model used. Options are "rpart", "party", "C50", or "caret".

Examples

# \donttest{
library(rpart)
library(partykit)
library(ggparty)
library(dplyr)
library(seriation)
data_all <- add_data_type(
  data_train = train_covid, data_test = test_covid
)
data <- prepare_features(
  data_all,
  target_lab = "Outcome",
  task = "classification"
)
train_tree <- train_tree(
  data_train = train_covid,
  target_lab = "Outcome", model = "rpart"
)
fit <- train_tree$fit
var_imp <- train_tree$var_imp
tree_res <- compute_tree(
  fit,
  model = "rpart", show = "test",
  data = data, target_lab = "Outcome",
  task = "classification"
)
prepare_tree(tree_res, model = "rpart")
# }