1. 学ぶ
  2. /
  3. コース
  4. /
  5. HR Analytics: Przewidywanie rotacji pracowników w Pythonie

Connected

演習

Ograniczanie minimalnej liczby próbek w liściu

Innym sposobem zapobiegania przeuczeniu jest określenie minimalnej liczby obserwacji wymaganej do utworzenia liścia (lub węzła) w drzewie decyzyjnym.

W tym ćwiczeniu:

  • ustawisz ten minimalny limit na 100
  • dopasujesz nowy model do danych pracowniczych
  • przeanalizujesz wyniki predykcji na zbiorze treningowym i testowym

Zmienne features_train, target_train, features_test i target_test są już dostępne w twoim środowisku.

指示

100 XP
  • Zainicjalizuj DecisionTreeClassifier i ustaw minimalny limit liścia na 100 obserwacji
  • Dopasuj model drzewa decyzyjnego do danych treningowych.
  • Sprawdź dokładność predykcji na zbiorze treningowym i testowym.