Dividindo a árvore
Dado o índice de Gini que resultaria ao dividir pela variável A ou B, respectivamente, decida por qual variável a árvore deve se dividir a seguir.
Este exercício faz parte do curso
HR Analytics: prevendo rotatividade de funcionários em Python
Instruções do exercício
- Verifique qual Gini é menor.
- Se A tiver Gini menor, divida por A; caso contrário, divida por B.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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!")