使用sklearn的决策树来分类鸢尾花数据时为啥生成的决策树PDF闪退?
from sklearn import tree
from sklearn.datasets import load_iris
iris=load_iris()
iris.data
iris.target
clf=tree.DecisionTreeClassifier()
clf=clf.fit(iris.data,iris.target)
import io
from io import StringIO
with open('iris.dot','w') as f:
f=tree.export_graphviz(clf,out_file=f)
import pydotplus
dot_data=tree.export_graphviz(clf,out_file=None)
graph=pydotplus.graph_from_dot_data(dot_data)
graph.write_pdf('iris.pdf')
True
在最后一行代码,明明显示了True,可是每次运行只有一个黑框一闪而过,决策树的PDF根本就没有出现,这是怎么回事呢?求问了解情况的UU。