Uczenie maszynowe w analizie marketingowej w R
Verena Pflieger
Data Scientist at INWT Statistics
1) Podział zbioru danych na dane treningowe i testowe
# Generating random index for training and test set
# set.seed ensures reproducibility of random components
set.seed(534381)
churnData$isTrain <- rbinom(nrow(churnData), 1, 0.66)
train <- subset(churnData, churnData$isTrain == 1)
test <- subset(churnData, churnData$isTrain == 0)
2) Budowa modelu na podstawie danych treningowych
# Modeling logitTrainNew
logitTrainNew <- glm( returnCustomer ~ title + newsletter +
websiteDesign + paymentMethod + couponDiscount +
purchaseValue + throughAffiliate +
shippingFees + dvd + blueray + vinyl +
videogameDownload + prodOthers + prodRemitted,
family = binomial, data = train)
# Out-of-sample prediction for logitTrainNew
test$predNew <- predict(logitTrainNew, type = "response",
newdata = test)
# Calculating the confusion matrix
confMatrixNew <- confusion.matrix(test$returnCustomer, test$predNew,
threshold = 0.3)
confMatrixNew
# Calculating the accuracy
accuracyNew <- sum(diag(confMatrixNew)) / sum(confMatrixNew)
accuracyNew
obs
pred 0 1
0 11939 2449
1 716 350
0.7951987

Obliczanie dokładności metodą kroswalidacji
library(boot)
# Accuracy function with threshold = 0.3
Acc03 <- function(r, pi = 0) {
cm <- confusion.matrix(r, pi, threshold = 0.3)
acc <- sum(diag(cm)) / sum(cm)
return(acc)}
# Accuracy
set.seed(534381)
cv.glm(churnData, logitModelNew, cost = Acc03, K = 6)$delta
0.7943894
| Wnioski – Regresja logistyczna | |
|---|---|
| Nauczyłeś/aś się... | jak przewidywać klientów sklepu internetowego skłonnych do odejścia |
| stosować binarną regresję logistyczną do obliczania prawdopodobieństw | |
| że wybór progu jest kluczowy |
| Wnioski z modelu | |
|---|---|
| Nauczyłeś/aś się... | że klienci zapisani na newsletter częściej wracają |
| że klienci korzystający z kuponu rzadziej wracają | |
| że klienci bez opłat za wysyłkę częściej wracają | |
| itp... |
Uczenie maszynowe w analizie marketingowej w R