我在Windows10上使用IPython来训练和绘制决策树。我知道以下代码在linux上运行了一段时间。我安装了pydot并且还有graphviz(正确指定了路径)。
# Train and draw out a decision tree
from IPython import display
from sklearn import datasets, tree, utils
from sklearn.externals.six import StringIO
import pydot
# Train a small decision tree on the iris dataset
dataset = datasets.load_iris()
X_iris, y_iris = utils.shuffle(dataset.data, dataset.target,random_state=42)
tree_clf = tree.DecisionTreeClassifier(max_depth=3).fit(X_iris, y_iris)
# Generate a plot of the decision tree
dot_data = StringIO()
tree.export_graphviz(tree_clf, out_file=dot_data)
graph = pydot.graph_from_dot_data(dot_data.getvalue())
display.Image(graph.create_png())
我收到以下错误:
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-5-3452aa5e9794> in <module>()
15 tree.export_graphviz(tree_clf, out_file=dot_data)
16 graph = pydot.graph_from_dot_data(dot_data.getvalue())
---> 17 display.Image(graph.create_png())
AttributeError: 'list' object has no attribute 'create_png'
答案 0 :(得分:1)
我通过将所有pydot
命令更改为pydotplus
来解决了这个问题。(包括import pydotplus
)可以使用!pip install pydotplus
来安装pydotplus包。
参考 https://github.com/scikit-learn/scikit-learn/pull/7342/files