Description Usage Arguments Author(s) References Examples
Plots a fancy RPart decision tree using the pretty rpart plotter.
1 | fancyRpartPlot(model, main="", sub, palettes, ...)
|
model |
an rpart object. |
main |
title for the plot. |
sub |
sub title for the plot. The default is a Rattle string with date, time and username. |
palettes |
a list of sequential palettes names. As supported by RColorBrewer::brewer.pal the available names are Blues BuGn BuPu GnBu Greens Greys Oranges OrRd PuBu PuBuGn PuRd Purples RdPu Reds YlGn YlGnBu YlOrBr YlOrRd. |
type |
the type of plot to generate (2). |
... |
additional arguments passed on to prp. |
Package home page: http://rattle.togaware.com
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | ## Use rpart to build a decision tree.
library(rpart)
## Set up the data for modelling.
set.seed(42)
ds <- weather
target <- "RainTomorrow"
risk <- "RISK_MM"
ignore <- c("Date", "Location", risk)
vars <- setdiff(names(ds), ignore)
nobs <- nrow(ds)
form <- formula(paste(target, "~ ."))
train <- sample(nobs, 0.7*nobs)
test <- setdiff(seq_len(nobs), train)
actual <- ds[test, target]
risks <- ds[test, risk]
# Build the model.
model <- rpart(form, data=ds[train, vars])
## Plot the model.
fancyRpartPlot(model)
## Choose different colours
fancyRpartPlot(model, palettes=c("Greys", "Oranges"))
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.