Predecir Clientes con Naive Bayes

Una empresa quiere llamar a sus clientes para ofrecerles un nuevo producto, pero quiere contactar unicamente a los clientes con mayor probabilidad de compra, para ahorrar en el costo por llamada.

Si se tienen DATOS HISTÓRICOS (de campañas anteriores o pruebas pilotos) de clientes que SI aceptaron y clientes que NO aceptaron comprar un producto cuando fueron contactados, y también se tienen registros de NUEVOS CLIENTES a los que se le quiere calcular la probabilidad de compra, teniendo algo como:

Puede utilizarse el método Bayes Ingenuo (o Naive Bayes)  con la técnica Maximo a Posteriori (MAP) para clasificar a los clientes según su probabilidad de compra. Para esto se multiplica la probabilidad de Compra=Si de cada atributo (EstadoCvivil,Profesion, etc.) y el resultado se multiplica por la probabilidad total de Compra=Si. Luego se hace lo mismo con la probabilidad para Compra=No de cada atributo y se multiplica por la totalidad de Compra=No. El resultado que tenga mayor valor (Compra=Si, ó Compra=No), será el elegido por tener mayor probabilidad de compra.


El cálculo de la probabilidad de cada atributo y para cada cliente nuevo seria algo como:

El resultado es el siguiente archivo csv:

Los archivos Datos_Historicos.csv y Nuevos_Clientes.csv pueden descargarse AQUI


La programación en R sería la siguiente:

# --------------------------------------------------------------------------------
#  Crea las tablas (ó data frame) Datos_Historicos y Nuevos_Clientes
Datos_Historicos <- read.csv("C:/Archivos/Datos_Historicos.csv") # tabla de entrenamiento
Nuevos_Clientes <- read.csv("C:/Archivos/Nuevos_Clientes.csv") # tabla de test
 
# --------------------------------------------------------------------------------
# Crea  Probabilidades con Datos_Historicos  y asigna predicción a Nuevos_Clientes
library(e1071)  # librerias de naiveBayes
Probabilidades <- naiveBayes(Compra ~., data=Datos_Historicos[-1]) # crea tabla Probabilidades
Prediccion <- predict (Probabilidades , Nuevos_Clientes[,-6]); # crea prediccion a datos nuevos
 
# --------------------------------------------------------------------------------
# Adiciona columna de Prediccion a la tabla Nuevos_Clientes
Nuevos_Clientes$Prediccion_de_Compra <- Prediccion   
 
# ---------------------------------------------------------------------------------
# Guarda la tabla de Nuevos_Clientes con la predicción:
write.csv(Nuevos_Clientes,"C:/Archivos/Predict_NuevosClientes.csv")