Plotting Decision trees
Dr. D’Agostino McGowan
1 / 10
Dr. Lucy D'Agostino McGowan
Plotting decision trees
There are several R packages that assist with tree plotting
rpart.plotpartykitrattle
2 / 10
Dr. Lucy D'Agostino McGowan
Where to find out more about packages
- Vignettes
- Journal Article (R Journal is great)
- Rstudio Community
- StackOverflow
3 / 10
Dr. Lucy D'Agostino McGowan
05:00
Learn about partykit
Google partykit to find out how more about it (hint: use key words like rstats and decision tree plots)
4 / 10
Dr. Lucy D'Agostino McGowan
rpart.plot
We're going to focus on rpart.plot, but feel free to try the others!
5 / 10
Dr. Lucy D'Agostino McGowan
rpart.plot
We're going to focus on rpart.plot, but feel free to try the others!
install.packages("rpart.plot")5 / 10
Dr. Lucy D'Agostino McGowan
rpart.plot
We're going to focus on rpart.plot, but feel free to try the others!
install.packages("rpart.plot")library(rpart.plot)5 / 10
Dr. Lucy D'Agostino McGowan
rpart.plot
tree_spec <- decision_tree( cost_complexity = 0.1, tree_depth = 10, mode = "regression") %>% set_engine("rpart")model <- fit(tree_spec, Salary ~ Hits + Years + PutOuts + RBI + Walks + Runs, data = baseball)6 / 10
Dr. Lucy D'Agostino McGowan
rpart.plot
tree_spec <- decision_tree( cost_complexity = 0.1, tree_depth = 10, mode = "regression") %>% set_engine("rpart")model <- fit(tree_spec, Salary ~ Hits + Years + PutOuts + RBI + Walks + Runs, data = baseball)rpart.plot(model$fit, roundint = FALSE)6 / 10
Dr. Lucy D'Agostino McGowan
rpart.plot
tree_spec <- decision_tree( cost_complexity = 0.1, tree_depth = 10, mode = "regression") %>% set_engine("rpart")model <- fit(tree_spec, Salary ~ Hits + Years + PutOuts + RBI + Walks + Runs, data = baseball)rpart.plot(model$fit, roundint = FALSE)7 / 10
Dr. Lucy D'Agostino McGowan
rpart.plot
tree_spec <- decision_tree( cost_complexity = 0.1, tree_depth = 10, mode = "regression") %>% set_engine("rpart")model <- fit(tree_spec, Salary ~ Hits + Years + PutOuts + RBI + Walks + Runs, data = baseball)rpart.plot(model$fit, roundint = FALSE)8 / 10
Dr. Lucy D'Agostino McGowan
rpart.plot
rpart.plot(model$fit, roundint = FALSE)
9 / 10
Dr. Lucy D'Agostino McGowan
10:00
Application Exercise
- Open your application exercise [10]
- Install
rpart.plot - Create a plot from your final decision tree
10 / 10