[FIXED] Sklearn decision tree plot does not appear

Issue

I am trying to follow scikit learn example on decision trees:

from sklearn.datasets import load_iris
from sklearn import tree
X, y = load_iris(return_X_y=True)
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, y)

When I try to plot the tree:

tree.plot_tree(clf.fit(iris.data, iris.target))

I get

NameError                                 Traceback (most recent call last)
<ipython-input-2-e72b33a93ee6> in <module>
----> 1 tree.plot_tree(clf.fit(iris.data, iris.target))

NameError: name 'iris' is not defined

Solution

iris doesn't exist if you don't assign it. Use this line to plot:

tree.plot_tree(clf.fit(X, y))

You already assigned the X and y of load_iris() to a variable so you can use them.

Additionally, make sure the graphviz library's bin folder is in PATH.

Answered By - Nicolas Gervais