Advertisement
Guest User

Untitled

a guest
Sep 21st, 2017
56
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.33 KB | None | 0 0
  1. from IPython.display import Image
  2. from sklearn.tree import export_graphviz
  3. import io
  4. import pydot
  5.  
  6. def show_tree(dt):
  7. dotfile = io.StringIO()
  8. export_graphviz(dt, out_file=dotfile)
  9. png = io.BytesIO()
  10. pydot.graph_from_dot_data(dotfile.getvalue()).write_png(png)
  11. return Image(data=png.getvalue(), format="png", embed=True)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement