Dela upp trädet
Givet Gini-indexet som skulle bli resultatet av att dela upp på antingen variabel A eller B, avgör du med vilken variabel trädet ska dela upp härnäst.
Den här övningen är en del av kursen
HR-analys: Förutsäg personalomsättning i Python
Övningsinstruktioner
- Kontrollera vilket Gini-värde som är lägst.
- Om A har lägre Gini, dela upp på A, annars dela upp på B.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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!")