Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from IPython.display import Image
- from sklearn.tree import export_graphviz
- import io
- import pydot
- def show_tree(dt):
- dotfile = io.StringIO()
- export_graphviz(dt, out_file=dotfile)
- png = io.BytesIO()
- pydot.graph_from_dot_data(dotfile.getvalue()).write_png(png)
- return Image(data=png.getvalue(), format="png", embed=True)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement