Dostrajanie klasyfikatora rotacji pracowników

HR Analytics: Przewidywanie rotacji pracowników w Pythonie

Hrant Davtyan

Assistant Professor of Data Science American University of Armenia

Przeuczenie

Obecność przeuczenia:

Dokładność na zbiorze treningowym: 100%
Dokładność na zbiorze testowym: 97,23%

Metody przeciwdziałania:

Ograniczenie maksymalnej głębokości drzewa
Ograniczenie minimalnej liczby próbek w liściach

Przycinanie drzewa

Ograniczenie głębokości

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

Ograniczenie próbek

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

Czas na ćwiczenia!

HR Analytics: Przewidywanie rotacji pracowników w Pythonie