計算 Gini 指數
決策樹演算法的目標是讓終端節點的區分盡可能純淨。Gini 指數是用來達成此目標的方法之一。它是根據各群組中樣本所佔比例計算而來。
已知留下與離職的人數，請計算該節點的 Gini 指數。
本練習屬於課程
HR 分析：用 Python 預測員工流失
練習說明
- 計算該節點的員工總數。
- 依各群組員工所佔比例計算 Gini 指數。
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)