8.27.5. sklearn.tree.export_graphviz

sklearn.tree.export_graphviz(decision_tree, out_file=None, feature_names=None)

Export a decision tree in DOT format.

This function generates a GraphViz representation of the decision tree, which is then written into out_file. Once exported, graphical renderings can be generated using, for example:

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

Parameters :

decision_tree : decision tree classifier

The decision tree to be exported to graphviz.

out : file object or string, optional (default=None)

Handle or name of the output file.

feature_names : list of strings, optional (default=None)

Names of each of the features.

Returns :

out_file : file object

The file object to which the tree was exported. The user is expected to close() this object when done with it.

Examples

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

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

>>> clf = clf.fit(iris.data, iris.target)
>>> import tempfile
>>> out_file = tree.export_graphviz(clf, out_file=tempfile.TemporaryFile())
>>> out_file.close()