tree_segment_prettify {citrus}

Tree Segment Prettify Function

Description

Returns a prettier version of the decision tree.

Usage

tree_segment_prettify(tree, char_length = 20, print_plot = FALSE)

Arguments

tree

The decision tree model to prettify

char_length

integer, the character limit before truncating categories and putting them into an "other" group

print_plot

logical, indicates whether to print the generated plot or not

Value

A formatted and "prettified" rpart.plot object. This plot object can be plotted using the rpart::prp function.