sklearn.tree.plot_tree

sklearn.tree.plot_tree()

Trazar un árbol de decisión.

Los recuentos de muestras que se muestran se ponderan con cualquier sample_weights que pueda haber.

La visualización se ajusta automáticamente al tamaño del eje. Utilice los argumentos figsize o dpi de plt.figure para controlar el tamaño de la representación.

Lee más en el Manual de usuario.

Nuevo en la versión 0.21.

Parámetros

decision_treeel regresor o clasificador del árbol de decisión

El árbol de decisión que se va a trazar.

max_depthint, default=None

La profundidad máxima de la representación. Si es None, el árbol se genera completamente.

feature_nameslista de cadenas, default=None

Nombres de cada una de las características. Si es None, se utilizarán nombres genéricos («X[0]», «X[1]», …).

class_nameslista de cadenas o booleanos, default=None

Nombres de cada una de las clases objetivo en orden numérico ascendente. Sólo es relevante para la clasificación y no es compatible con la salida múltiple. Si es True, muestra una representación simbólica del nombre de la clase.

label{“all”, “root”, “none”}, default=”all”

Si se muestran etiquetas informativas para la impureza, etc. Las opciones incluyen “all” para mostrar en cada nodo, “root” para mostrar sólo en el nodo raíz superior, o “none” para no mostrar en ningún nodo.

filledbool, default=False

Cuando se establece en ``True`, pinta los nodos para indicar la clase mayoritaria para la clasificación, la extremidad de los valores para la regresión, o la pureza del nodo para la salida múltiple.

impuritybool, default=True

Cuando se establece en True, mostrar la impureza en cada nodo.

node_idsbool, default=False

Cuando se establece en True, mostrar el número de identificación en cada nodo.

proportionbool, default=False

Cuando se establece como True, cambia la visualización de valores y/o muestras para que sean proporciones y porcentajes respectivamente.

rotatebool, default=False

Este parámetro no tiene ningún efecto en la visualización de árboles de matplotlib y se mantiene aquí por compatibilidad con versiones anteriores.

Obsoleto desde la versión 0.23: rotate está obsoleto en 0.23 y será eliminado en 1.0 (cambio de nombre de 0.25).

roundedbool, default=False

Si se establece como True, dibujar las cajas de los nodos con esquinas redondeadas y utilizar fuentes Helvetica en lugar de Times-Roman.

precisionint, default=3

Número de dígitos de precisión para el punto flotante en los valores de los atributos impureza, umbral y valor de cada nodo.

axmatplotlib axis, default=None

Ejes para trazar. Si es None, se utiliza el eje actual. Cualquier contenido anterior se borra.

fontsizeint, default=None

Tamaño de la fuente del texto. Si es None, se determina automáticamente para ajustarse a la figura.

Devuelve

annotationslista de artistas

Lista que contiene los artistas de las cajas de anotación que componen el árbol.

Ejemplos

>>> from sklearn.datasets import load_iris
>>> from sklearn import tree

>>> clf = tree.DecisionTreeClassifier(random_state=0)
>>> iris = load_iris()

>>> clf = clf.fit(iris.data, iris.target)
>>> tree.plot_tree(clf)  
[Text(251.5,345.217,'X[3] <= 0.8...