Plot a sklearn decision tree

plot_sklearn_decision_tree.py

import pydot

from IPython.display import Image
from sklearn.externals.six import StringIO
from sklearn.tree import export_graphviz

dot_data = StringIO()  
export_graphviz(cl, out_file=dot_data,  
                    feature_names=Xtrain.columns,  
                    class_names=['class0','class1'],  
                    filled=True, rounded=True,  
                    special_characters=True)
graph = pydot.graph_from_dot_data(dot_data.getvalue())[0]  
Image(graph.create_png())