Rozdzielanie zmiennej docelowej i cech
Aby dokonać predykcji (w tym przypadku: czy pracownik odejdzie z firmy, czy nie), należy podzielić zbiór danych na dwa komponenty:
- zmienna zależna lub zmienna docelowa – wartość, którą chcemy przewidzieć
- zmienne niezależne lub cechy – dane wykorzystywane do dokonania predykcji
Twoim zadaniem jest wyodrębnienie target i features. Zmienną docelową jest tutaj wskaźnik odejść pracowników (churn), a cechy obejmują wszystko pozostałe.
Przypomnienie: zbiór danych został już wcześniej zmodyfikowany – zakodowano zmienne kategoryczne i zastosowano kodowanie binarne (dummies).
Biblioteka pandas została zaimportowana jako pd.
To ćwiczenie jest częścią kursu
HR Analytics: Przewidywanie rotacji pracowników w Pythonie
Instrukcje do ćwiczenia
- Ustaw zmienną docelową i cechy:
- Wybierz kolumnę ze zmienną zależną (
churn) i przypisz ją dotarget. - Użyj
.drop(), aby usunąć kolumnęchurni przypisać wszystko pozostałe dofeatures.
- Wybierz kolumnę ze zmienną zależną (
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Set the target and features
# Choose the dependent variable column (churn) and set it as target
target = data.____
# Drop column churn and set everything else as features
features = data.____("____",axis=1)