Podział drzewa
Na podstawie indeksu Giniego obliczonego dla podziału według zmiennej A lub B zdecyduj, według której zmiennej drzewo powinno zostać podzielone w następnym kroku.
To ćwiczenie jest częścią kursu
HR Analytics: Przewidywanie rotacji pracowników w Pythonie
Instrukcje do ćwiczenia
- Sprawdź, który indeks Giniego jest niższy.
- Jeśli A ma niższy indeks Giniego, dokonaj podziału według A; w przeciwnym razie – według B.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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!")