1. Learn
    2. /
  2. Курси
    3. /
  3. HR Analytics: прогнозування плинності персоналу в Python
Connected

Вправа

Обмеження розміру вибірки

Інший спосіб уникнути перенавчання — задати мінімальну кількість спостережень, необхідних для розростання листка (або вузла) у рішальному дереві.

У цій вправі ви:

  • встановите цей мінімальний поріг у 100
  • навчите нову модель на даних про співробітників
  • перевірите результати передбачення на тренувальній і тестовій вибірках

Змінні features_train, target_train, features_test і target_test вже доступні у вашому робочому середовищі.

Інструкції

100 XP
  • Ініціалізуйте DecisionTreeClassifier і встановіть мінімальний поріг для листка у 100 спостережень
  • Навчіть модель рішального дерева на тренувальних даних.
  • Перевірте точність передбачень як на тренувальній, так і на тестовій вибірках.