1.首先我调用graphviz画决策树,代码如下,报错如下
from sklearn.datasets import load_iris
from sklearn import tree
#import pydotplus
import graphviz
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
dot_data =tree.export_graphviz(clf,out_file=None,
feature_names=iris.feature_names,
class_names=iris.target_names,
filled=True, rounded=True,
special_characters=True)
graph=graphviz.Source(dot_data)
graph
报错如下:
2.我添加了pydotplus 包,这次不报错了,但是无法显示图形,可以下载pdf
代码如下:
结果如下:
希望有遇到过此类问题的伙伴帮忙解答!