1. Nauka
  2. /
  3. Kursy
  4. /
  5. HR Analytics: Przewidywanie rotacji pracowników w Pythonie

Connected

ćwiczenie

Porównanie modeli przewidywania rotacji pracowników

W tym ćwiczeniu porównasz modele zbalansowany i niezbalansowany (domyślny), korzystając z przyciętego drzewa (max_depth=7). Model niezbalansowany został już oceniony przy użyciu miar recall i ROC/AUC. Wykonaj te same kroki dla modelu zbalansowanego.

  • Zmienne features_train, target_train, features_test i target_test są już dostępne w twoim środowisku pracy.
  • Model niezbalansowany został już dopasowany, a jego predykcje zapisano jako prediction.
  • Funkcje recall_score() i roc_auc_score() zostały już zaimportowane.

Instrukcje

100 XP
  • Zainicjalizuj model zbalansowany, ustawiając jego maksymalną głębokość na 7 i ziarno losowości na 42.
  • Dopasuj go do danych treningowych.
  • Wygeneruj predykcje na zbiorze testowym.
  • Wyświetl wartość miary recall oraz wynik ROC/AUC.