De boom splitsen
Gezien de Gini-index die zou ontstaan door te splitsen op variabele A of B, bepaal op welke variabele de boom als volgende moet splitsen.
Deze oefening maakt deel uit van de cursus
HR-analytics: verloop van medewerkers voorspellen in Python
Oefeninstructies
- Controleer welke Gini lager is.
- Als A een lagere Gini heeft, split dan op A; anders split je op B.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# 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!")