Suddividere l'albero
Dato l'indice di Gini che risulterebbe dallo split sulla variabile A o sulla variabile B, decidi in base a quale variabile l'albero dovrebbe effettuare il prossimo split.
Questo esercizio fa parte del corso
HR Analytics: prevedere l'abbandono dei dipendenti in Python
Istruzioni dell'esercizio
- Verifica quale Gini è più basso.
- Se A ha un Gini più basso, fai lo split su A; altrimenti, fai lo split su B.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# 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!")