`sklearn.tree`.export_graphviz¶

sklearn.tree.export_graphviz()¶

Exporta un árbol de decisiones en formato DOT.

Esta función genera una representación GraphViz del árbol de decisión, que se escribe en out_file. Una vez exportado, se pueden generar representaciones gráficas utilizando, por ejemplo:

$ dot -Tps tree.dot -o tree.ps      (PostScript format)
$ dot -Tpng tree.dot -o tree.png    (PNG format)

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

Lee más en el Manual de usuario.

Parámetros

decision_treeclasificador de árbol de decisión: El árbol de decisiones que a ser exportado a GraphViz.
out_fileobjeto o str, default=None: Manejador o nombre del archivo de salida. Si None, el resultado se devuelve como una cadena.

Distinto en la versión 0.20: Se ha cambiado el valor por defecto de out_file de «tree.dot» a None.
max_depthint, default=None: La profundidad máxima de la representación. Si es None, el árbol se genera completamente.
feature_nameslista de str, default=None: Nombres de cada una de las características. Si es None los nombres genéricos serán usados («feature_0», «feature_1», …).
class_nameslista de str o bool, 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.
leaves_parallelbool, default=False: Cuando se establece a True, dibuja todos los nodos de hoja en la parte inferior del árbol.
impuritybool, default=True: Cuando se establece en True, muestra la impureza en cada nodo.
node_idsbool, default=False: Cuando se establece en True, muestra el número ID 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: Cuando se establece en True, orienta el árbol de izquierda a derecha en lugar de arriba.
roundedbool, default=False: Si se establece como True, dibuja las cajas de los nodos con esquinas redondeadas y utiliza fuentes Helvetica en lugar de Times-Roman.
special_charactersbool, default=False: Si se establece como False, se ignoran los caracteres especiales por compatibilidad con PostScript.
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.

Devuelve

dot_datacadena: Representación del árbol de entrada en formato dot de GraphViz. Sólo se devuelve si out_file es None.

Nuevo en la versión 0.18.

Ejemplos

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

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

>>> clf = clf.fit(iris.data, iris.target)
>>> tree.export_graphviz(clf)
'digraph Tree {...

sklearn.tree.export_graphviz¶

`sklearn.tree`.export_graphviz¶