Visualize simple 2-D decision tree partitions in R. The parttree package is optimised to work with ggplot2, although it can be used to visualize tree partitions with base R graphics too.

## Installation

This package is not yet on CRAN, but can be installed from GitHub with:

# install.packages("remotes")
remotes::install_github("grantmcdermott/parttree")

## Quickstart

The parttree homepage includes an introductory vignette and detailed documentation. But here’s a quickstart example using the “kyphosis” dataset that comes bundled with the rpart package. In this case, we are interested in predicting kyphosis recovery after spinal surgery, as a function of 1) the number of topmost vertebra that were operated, and 2) patient age. The key visualization layer below—provided by this package—is geom_partree().

library(rpart)     # For the dataset and fitting decisions trees
library(parttree)  # This package (will automatically load ggplot2 too)
theme_minimal()