Plot a CART Decision Tree from sklearn with pydotplus and graphviz

plot_tree.py

def plot_tree():
  """Make sure you have the correct imports.
  Imports:
      from sklearn import tree
      import pydotplus 
      from IPython.display import Image 
  
  Note: Make sure you already have a 'tree'.
  ...
  clf = tree.DecisionTreeClassifier()
  clf = clf.fit(X, y)
  ...
  """
  dot_data = tree.export_graphviz(clf, out_file=None)
  graph = pydotplus.graph_from_dot_data(dot_data)  
  return Image(graph.create_png())