การคำนวณ Gini index
อัลกอริทึม Decision Tree มีเป้าหมายเพื่อแบ่งข้อมูลในโหนดปลายทางให้มีความบริสุทธิ์มากที่สุด Gini index เป็นหนึ่งในวิธีที่ใช้เพื่อบรรลุเป้าหมายนี้ โดยคำนวณจากสัดส่วนของตัวอย่างในแต่ละกลุ่ม
เมื่อทราบจำนวนพนักงานที่อยู่ต่อและพนักงานที่ลาออก ให้คำนวณ Gini index ของโหนดนั้น
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
HR Analytics: การพยากรณ์การลาออกของพนักงานด้วย Python
คำแนะนำการฝึกหัด
- คำนวณจำนวนพนักงานทั้งหมดในโหนดนั้น
- คำนวณ Gini index จากสัดส่วนของพนักงานในแต่ละกลุ่ม
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)