Tính chỉ số Gini
Thuật toán decision tree hướng tới việc tạo ra các nút lá có độ “thuần” cao nhất có thể. Chỉ số Gini là một trong các phương pháp được dùng để đánh giá điều này. Nó được tính dựa trên tỷ lệ mẫu trong mỗi nhóm.
Cho số người lần lượt ở lại và rời đi, hãy tính chỉ số Gini cho nút đó.
Bài tập này là một phần của khóa học
Phân tích Nhân sự (HR Analytics): Dự đoán nghỉ việc bằng Python
Hướng dẫn bài tập
- Tính tổng số nhân viên trong nút đó.
- Tính chỉ số Gini dựa trên tỷ lệ nhân viên ở mỗi nhóm.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)