Plotting Decision trees
Dr. D’Agostino McGowan
1 / 11
Dr. Lucy D'Agostino McGowan
05:00
📖 Take 06 Assessment in Canvas
2 / 11
Dr. Lucy D'Agostino McGowan
Plotting decision trees
There are several R packages that assist with tree plotting
rpart.plotpartykitrattle
3 / 11
Dr. Lucy D'Agostino McGowan
Where to find out more about packages
- Vignettes
- Journal Article (R Journal is great)
- Rstudio Community
- StackOverflow
4 / 11
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)
5 / 11
Dr. Lucy D'Agostino McGowan
rpart.plot
We're going to focus on rpart.plot, but feel free to try the others!
6 / 11
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")6 / 11
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)6 / 11
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)7 / 11
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 / 11
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 / 11
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)9 / 11
Dr. Lucy D'Agostino McGowan
rpart.plot
rpart.plot(model$fit, roundint = FALSE)
10 / 11
Dr. Lucy D'Agostino McGowan
10:00
AE 05 - Regression trees
- Open your application exercise (05) from last week
- Install
rpart.plot - Create a plot from your final decision tree
- Knit, Commit, Push. Be sure to have the final results pushed to GitHub by April 8 at noon
11 / 11