Sklearn决策树图未出现

Question

我正在尝试在决策树上遵循scikit学习示例：

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)

当我尝试绘制树时：

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

我明白了

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

Answer 1

iris如果不分配，则不存在。使用此线来绘制：

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

您已经将X的{​​{1}}和y分配给了一个变量，以便您可以使用它们。

另外，请确保load_iris()库的graphviz文件夹位于bin中。