Chia nhánh cây
Với chỉ số Gini thu được khi tách theo biến A hoặc B tương ứng, hãy quyết định cây nên tách tiếp theo theo biến nào.
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
- Kiểm tra Gini nào nhỏ hơn.
- Nếu A có Gini nhỏ hơn, tách theo A; nếu không, tách theo B.
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.
# Gini index in case of splitting by variable A or B
gini_A = 0.65
gini_B = 0.15
# check which Gini is lower and use it for spliting
if ____ < ____:
print("split by A!")
____:
print("split by B!")