調整員工流失分類器

HR 分析：用 Python 預測員工流失

Hrant Davtyan

Assistant Professor of Data Science American University of Armenia

過度擬合

過度擬合的跡象：

  • 訓練準確率：100%
  • 測試準確率：97.23%

對策：

  • 限制樹的最大深度
  • 限制葉節點的最小樣本數
HR 分析：用 Python 預測員工流失

剪枝決策樹

限制深度

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 預測員工流失

一起來練習吧！

HR 分析：用 Python 預測員工流失

Preparing Video For Download...