Налаштування класифікатора плинності кадрів

HR Analytics: прогнозування плинності персоналу в 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 Analytics: прогнозування плинності персоналу в Python