1. Learn
    2. /
  2. Courses
    3. /
  3. HR Analytics: Predicting Employee Churn in Python

Exercise

Separating Target and Features

In order to make a prediction (in this case, whether an employee would leave or not), one needs to separate the dataset into two components:

  • the dependent variable or target which needs to be predicted
  • the independent variables or features that will be used to make a prediction

Your task is to separate the target and features. The target you have here is the employee churn, and features include everything else.

Reminder: the dataset has already been modified by encoding categorical variables and getting dummies.

pandas has been imported for you as pd.

Instructions

100 XP
  • Set the target and features:
    • Choose the dependent variable column (churn) and set it as target.
    • .drop() the column churn to set everything else as features.