Větvení stromu
Na základě Giniho indexu, který by vznikl rozdělením podle proměnné A nebo B, rozhodni, podle které proměnné má strom v dalším kroku větvit.
Toto cvičení je součástí kurzu
HR Analytics: Predicting Employee Churn in Python
Pokyny k cvičení
- Zjisti, který Giniho index je nižší.
- Pokud má A nižší Giniho index, větvi podle A, jinak větvi podle B.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# 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!")