sklearn.tree.plot_tree show returns chunk of text instead of visualised tree

Question

I'm trying to show a tree visualisation using plot_tree, but it shows a chunk of text instead:

from sklearn.tree import plot_tree
plot_tree(t)

(where t is an instance of DecisionTreeClassifier) This is the output:

[Text(464.99999999999994, 831.6, 'X[3] <= 0.8\nentropy = 1.581\nsamples = 120\nvalue = [39, 37, 44]'),
 Text(393.46153846153845, 646.8, 'entropy = 0.0\nsamples = 39\nvalue = [39, 0, 0]'),

and so on and so forth. How do I make it show the visual tree instead? I'm using Jupyter 6.4.1 and I already imported matplotlib earlier in the code. Thanks!

Alex Serra Marrugat · Accepted Answer · 2021-10-27 07:27:20Z

7

In my case, it works with a simple "show":

plot_tree(t)
plt.show()

edited Oct 27, 2021 at 7:27

answered Oct 27, 2021 at 6:27

Alex Serra Marrugat

2,0421 gold badge6 silver badges15 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Mohammad Amin Dadgar · Accepted Answer · 2022-05-20 05:22:22Z

0

You can plot your tree and specify the plot size of your tree with plt.figure

width = 10
height = 7
plt.figure(figsize=(width, height))

tree_plot_max_depth = 6
plot_tree(t, max_depth=tree_plot_max_depth)

## the key to the problem of not showing tree is the command below
plt.show()

And as the documentation is mentioned below you can specify more parameters for your tree to get a more informative image.

https://scikit-learn.org/stable/modules/generated/sklearn.tree.plot_tree.html

answered May 20, 2022 at 5:22

Mohammad Amin Dadgar

212 silver badges4 bronze badges

Collectives™ on Stack Overflow

sklearn.tree.plot_tree show returns chunk of text instead of visualised tree

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related