分割決策樹
已知分別以變數 A 或 B 進行分割所得到的 Gini 指數,請判斷下一步應該用哪一個變數來分割決策樹。
本練習屬於課程
HR 分析:用 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!")