The following functions allow visualization of various aspects of GBM gradient boosted regression tree outputs
tree_plot
- Creates a tree of histograms showing partitioning of data through a single tree within the model
- Only able to plot 6 splits Example:
sankey_tree
- Uses a sankey diagram to visualize data partitioning through a single tree in the model
- Can view all splits Example:
aggregate_split_sankey
- Aggregates all splits and following splits to show the most common split variables and the following split
- Option to exclude terminal nodes Example:
aggregate_tree_sankey