Настройка классификатора текучести кадров

HR-аналитика: прогнозирование текучести кадров на Python

Hrant Davtyan

Assistant Professor of Data Science American University of Armenia

Переобучение

Признаки переобучения:

Точность на обучающей выборке: 100%
Точность на тестовой выборке: 97,23%

Методы борьбы с ним:

Ограничение максимальной глубины дерева
Ограничение минимального размера выборки в листьях

Обрезка дерева

Ограничение глубины

model_depth_5 = DecisionTreeClassifier(
                max_depth=5, random_state=42)
# Train set Accuracy: 97.71%
# Test  set Accuracy: 97.06%

Ограничение выборки

model_sample_100 = DecisionTreeClassifier(
                   min_samples_leaf=100, random_state=42)
# Train set Accuracy: 96.58%
# Test set Accuracy: 96.13%

Давайте потренируемся!

HR-аналитика: прогнозирование текучести кадров на Python