Session Ready
Exercise

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.

Instructions 1/2
undefined XP
  • 1
  • 2
  • Split your data into 75% training and 25% testing using the initial_split() function.
  • Extract the training and testing dataframes from data_split using training() and testing(), respectively.