node_depth {bartMan} | R Documentation |
Calculate Node Depths in a Tree Data Frame
Description
Computes the depth of each node in a given tree data frame, assuming a binary tree structure. Requires the tree data frame to contain a logical column 'terminal' indicating terminal nodes.
Usage
node_depth(tree)
Arguments
tree |
A data frame representing a tree, must contain a 'terminal' column. |
Value
A vector of depths corresponding to each node in the tree.
Examples
data("tree_data_example")
# Create Terminal Column
tree_data_example <- transform(tree_data_example, terminal = ifelse(is.na(var), TRUE, FALSE))
# Get depths
depthList <- lapply(split(tree_data_example, ~treeNum + iteration),
function(x) cbind(x, depth = node_depth(x)-1))
# Turn into data frame
tree_data_example <- dplyr::bind_rows(depthList, .id = "list_id")
[Package bartMan version 0.1.1 Index]