การแบ่งต้นไม้

จากค่า Gini index ที่ได้จากการแบ่งด้วยตัวแปร A และ B ตามลำดับ ให้ตัดสินใจว่าควรแบ่งต้นไม้ครั้งถัดไปด้วยตัวแปรใด

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

HR Analytics: การพยากรณ์การลาออกของพนักงานด้วย Python

คำแนะนำการฝึกหัด

ตรวจสอบว่าค่า Gini ใดต่ำกว่ากัน
ถ้า A มีค่า Gini ต่ำกว่า ให้แบ่งด้วย A มิฉะนั้นให้แบ่งด้วย B

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

# 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!")

แก้ไขและรันโค้ด