開始使用免費開始

分割決策樹

已知分別以變數 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!")
編輯並執行程式碼