Fork me on GitHub

sklearn.tree.export_graphviz

sklearn.tree.export_graphviz(decision_tree, out_file='tree.dot', feature_names=None, max_depth=None, close=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 : file object or string, optional (default=”tree.dot”)

Handle or name of the output file.

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

Names of each of the features.

max_depth : int, optional (default=None)

The maximum depth of the representation. If None, the tree is fully generated.

Examples

>>> 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,
...     out_file='tree.dot')                
Previous
Next