Understanding the model predictions
In this exercise, you will predict and visualize the probability of turnover for all the cases in training and test datasets.
The model (final_log
), and train_set
and test_set
datasets are available in your workspace.
This exercise is part of the course
HR Analytics: Predicting Employee Churn in R
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Make predictions for training dataset
prediction_train <- ___(final_log, newdata = train_set,
type = "response")
# Look at the prediction range
___