plot.DecisionTreeObject: plotDecisionTree

In tomis9/decisionTree: Deciosion tree algorithm

Description

plot a decision tree

Usage

## S3 method for class 'DecisionTreeObject'
plot(DecisionTreeObject, dict, green = 1,
  left = 1)

Arguments

DecisionTreeObject	object of class DecisionTreeObject generated by function decisionTree
dict	a data.frame of optional names for nodes, colnames of data.frame are: vName and toPlot, remember about stringsAsFactors=F
green	which class should be coloured in green? 0 or 1?
left	which class should be mentioned on leaves

Examples

d <- iris[, c("Species", "Sepal.Length", "Sepal.Width")]
d$Species <- as.character(d$Species)
d$Species[d$Species != "setosa"] <- "non-setosa"
x <- d$Sepal.Length
x[d$Sepal.Length <= 5.2] <- "Very Short"
x[d$Sepal.Length >  5.2 & d$Sepal.Length <= 6.1] <- "Short"
x[d$Sepal.Length >  6.1 & d$Sepal.Length <= 7.0] <- "Long"
x[d$Sepal.Length >  7.0] <- "Very Long"
d$Sepal.Length <- x
decTree <- decisionTree(d, eta=5, purity=0.95, minsplit=0)
plot(decTree)

tomis9/decisionTree documentation built on May 29, 2019, 9:55 a.m.