Prepare train-test-validate parts
In this exercise, you will leverage the tools that you have learned thus far to build a classification model to predict employee attrition.
You will work with the attrition dataset, which contains 30 features about employees which you will use to predict if they have left the company.
You will first prepare the training & testing data sets, then you will further split the training data using cross-validation so that you can search for the best performing model for this task.
This exercise is part of the course
Machine Learning in the Tidyverse
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
set.seed(42)
# Prepare the initial split object
data_split <- initial_split(___, prop = ___)
# Extract the training data frame
training_data <- training(___)
# Extract the testing data frame
testing_data <- testing(___)