Árboles de clasificación, parte 2

Activando las librerías

library(tree)

library(rpart)

library(rpart.plot)

Predicciones con árboles de clasificación

arbol <- rpart(Species ~ Petal.Length + Petal.Width, data = iris, method = "class")
arbol

## n= 150 
## 
## node), split, n, loss, yval, (yprob)
##       * denotes terminal node
## 
## 1) root 150 100 setosa (0.33333333 0.33333333 0.33333333)  
##   2) Petal.Length< 2.45 50   0 setosa (1.00000000 0.00000000 0.00000000) *
##   3) Petal.Length>=2.45 100  50 versicolor (0.00000000 0.50000000 0.50000000)  
##     6) Petal.Width< 1.75 54   5 versicolor (0.00000000 0.90740741 0.09259259) *
##     7) Petal.Width>=1.75 46   1 virginica (0.00000000 0.02173913 0.97826087) *

Graficando el árbol de clasificación: función rpart.plot(), 1

rpart.plot(arbol)

Graficando el árbol de clasificación: función rpart.plot(), 2

rpart.plot(arbol, box.palette="RdBu", shadow.col="gray", nn=TRUE)

Graficando el árbol de clasificación: función rpart.plot(), 3

rpart.plot(arbol, type = 4, extra = 0, branch.lty = 3, box.palette = "RdYlGn")

Graficando el árbol de clasificación: función prp(), 1

prp(arbol)

Graficando el árbol de clasificación: función prp(), 2

prp(arbol, branch.type = 5, yesno = FALSE, faclen = 0)

Creando los datos que se van a clasificar

unseen <- data.frame(Petal.Length = c(1.7, 5.4),
                     Petal.Width = c(0.8, 2.3))
unseen

##   Petal.Length Petal.Width
## 1          1.7         0.8
## 2          5.4         2.3

Predicción con árbol de clasificación: función predict()

predict(arbol, unseen) # Probabilidad

##   setosa versicolor virginica
## 1      1 0.00000000 0.0000000
## 2      0 0.02173913 0.9782609

Clasificando los datos segun la predicción con árbol de clasificación

predict(arbol, unseen, type = "class") # Especie

##         1         2 
##    setosa virginica 
## Levels: setosa versicolor virginica

Este script fue generado mediante R en January 17, 2022 y forma parte de las actividades realizadas en las materias de Estadística y Taller IV. Facultad de Economía, Ciudad Universitaria. Contacto: cghv94@outlook.es y @cghv94. Esta obra está bajo una licencia de Creative Commons, Attribution 4.0 International (CC BY 4.0). Creative Commons (CC).