Create a labeled inner-panel generator for tree plots

Builds a partykit graph-content generator that draws internal tree nodes with readable variable labels and optional split-test p-values. The returned generator is intended for use in the inner_panel argument of partykit::plot.party().

Usage

tree_inner_panel_labeled(
  obj,
  var_labels = NULL,
  id = FALSE,
  show_p = TRUE,
  gp = grid::gpar(),
  fill = "white"
)

Arguments

obj

A fitted tree object inheriting from party.

var_labels

An optional named character vector of display labels.

id

Currently unused.

show_p

Logical; whether to display split-test p-values in internal nodes.

gp

A grid::gpar() object applied to the node viewport.

fill

Fill color used for the rounded internal node box.

Value

A graph-content generator function suitable for the inner_panel argument of partykit::plot.party().

Examples

data(kenya, package = "ineqTrees")
kenya_plot_vars <- c("wealth", "deadu5_num", "rural", "ed", "reg", "unskilled")
kenya_plot_data <- kenya[
  stats::complete.cases(kenya[, kenya_plot_vars]),
  kenya_plot_vars
]
set.seed(20260512)
kenya_plot_data <- kenya_plot_data[
  sample.int(nrow(kenya_plot_data), 800L),
  ,
  drop = FALSE
]
kenya_plot_fit <- ci_tree(
  cbind(wealth, deadu5_num) ~ rural + ed + reg + unskilled,
  data = kenya_plot_data,
  rank_name = "wealth",
  outcome_name = "deadu5_num",
  control = ci_tree_control(
    minsplit = 100,
    minbucket = 50,
    minprob = 0.05,
    maxdepth = 3
  )
)
kenya_var_labels <- c(
  rural = "Residence",
  ed = "Mother education",
  reg = "Province",
  unskilled = "Mother occupation"
)

inherits(tree_inner_panel_labeled, "grapcon_generator")
#> [1] TRUE
is.function(tree_inner_panel_labeled(
  kenya_plot_fit,
  var_labels = kenya_var_labels
))
#> [1] TRUE