Ocultar el nombre del criterio en la función plot_tree().
Necesito que el gráfico de la función plot_tree() se muestre sin un valor de criterio en cada nodo del árbol. Es decir, quiero obtener una imagen como la siguiente:
https://i.stack.imgur.com/ufvCa.png
sin usar Paint 🙂 ¿Cómo puedo hacerlo?
También necesito reemplazar la palabra ‘value’ por ‘proba’, ¿se puede hacer cambiando solo el código fuente?
davy.ai
Para eliminar el valor del criterio en cada nodo del árbol de decisión generado por la función
plot_tree()
en scikit-learn, puedes establecer el parámetrorounded
enTrue
. Esto eliminará la visualización del valor del criterio y hará que los bordes sean más curvos, lo que resultará en una imagen más clara del árbol de decisión.Aquí hay un ejemplo de código:
Para reemplazar la palabra “valor” por “proba”, tendrás que cambiar el código fuente de la función
plot_tree()
. Específicamente, deberás modificar el método_tree.plot_tree()
ubicado en el archivo_tree.pyx
dentro del paquete scikit-learn.Sin embargo, no es recomendable modificar el código fuente directamente ya que puede haber problemas de compatibilidad y mantenimiento. En su lugar, puedes crear una nueva función o subclase de la clase
DecisionTreeClassifier
, y anular el métodoplot_tree()
para mostrar “proba” en lugar de “valor”.