Ahora que ya conoces las entropías del nodo raíz y de los nodos hijos, puedes calcular la ganancia de información que aporta el color.

En los ejercicios anteriores calculaste entropy_root, entropy_left y entropy_right. Están disponibles en la consola.

Recuerda que vas a tomar la media ponderada de las entropías de los nodos hijos. Por tanto, tendrás que calcular qué proporción de las observaciones originales terminó en el lado izquierdo y derecho del corte. Guarda esos valores en p_left y p_right, respectivamente.

decison tree split by color

Información e importancia de las características

Create Your Free Account