Împărțirea arborelui
Pe baza indicelui Gini obținut prin împărțirea după variabila A sau după variabila B, decide după care dintre ele ar trebui să se facă următoarea ramificare a arborelui.
Acest exercițiu face parte din cursul
HR Analytics: Predicția fluctuației angajaților în Python
Instrucțiuni pentru exercițiu
- Verifică care valoare Gini este mai mică.
- Dacă A are un indice Gini mai mic, împarte după A; altfel, împarte după B.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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!")