Dividir el árbol
Dado el índice de Gini que resultaría de dividir por la variable A o por la B, respectivamente, decide por qué variable debería dividirse el árbol a continuación.
Este ejercicio forma parte del curso
Analítica de RR. HH.: Predicción de rotación de empleados en Python
Instrucciones del ejercicio
- Comprueba qué Gini es menor.
- Si A tiene un Gini menor, divide por A; en caso contrario, divide por B.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# 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!")