Está en la página 1de 1

Cdigo para el ejemplo practico

# Predecir Perdida de Clientes con Arbol de Decision

# PASO 1: Carga Package y Set de datos


# ---------------------------------------------------------------------------
library(C50)
library(rpart)
library(rpart.plot)
data(churn); # carga tablas

Variables <-c(4,7,16,19,17,20) # variables elegidas


Entrenamiento <-churnTrain[,Variables] # tabla entrenamiento
Test <-churnTest [,Variables] # tabla Test

# PASO 2: Crea Arbol de Decision


# ---------------------------------------------------------------------------
ModeloArbol<-rpart(churn ~ .,data=Entrenamiento,parms=list(split="information"))

# PASO 3: Predice Desafiliacin en datos de TEST


# ---------------------------------------------------------------------------
Prediccion <- predict(ModeloArbol, Test,type="class") # Predicccin en Test
MC <- table(Test[, "churn"],Prediccion) # Matriz de Confusin

# PASO 4: Crea Grafico


# ---------------------------------------------------------------------------
rpart.plot(ModeloArbol, type=1, extra=100,cex = .7,
box.col=c("gray99", "gray88")[ModeloArbol$frame$yval])

También podría gustarte