Predicting churn
In this exercise, you will use the predict() function in the pROC package to predict the churn probability of the customers in the test set, test_set.
The function has three arguments:
- The model used to make the predictions.
newdata: The dataset the model should be applied to.type: the type of prediction. We want to know the probability of churn, so we choose eitherresponseorprob.
Use the function predict() to predict the churn probability of the customers in test_set.
Cet exercice fait partie du cours
Predictive Analytics using Networked Data in R
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Load the package
library(pROC)
# Predict with the first model
firstPredictions <- predict(___, newdata = test_set, type = "response")