Tinh chỉnh bộ phân loại nghỉ việc của nhân viên

Phân tích Nhân sự (HR Analytics): Dự đoán nghỉ việc bằng Python

Hrant Davtyan

Assistant Professor of Data Science American University of Armenia

Overfitting

Dấu hiệu overfitting:

Độ chính xác huấn luyện: 100%
Độ chính xác kiểm thử: 97.23%

Cách khắc phục:

Giới hạn độ sâu tối đa của cây
Giới hạn kích thước mẫu tối thiểu ở lá

Cắt tỉa cây quyết định

Giới hạn độ sâu

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

Giới hạn số mẫu

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

Cùng luyện tập nào!

Phân tích Nhân sự (HR Analytics): Dự đoán nghỉ việc bằng Python