用jupyter lab 写决策树,调用graphviz报错,调用pydotplus后不报错了,但是无法显示图形

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

报错如下:

img

img

img

2.我添加了pydotplus 包,这次不报错了,但是无法显示图形,可以下载pdf
代码如下:
结果如下:

img

希望有遇到过此类问题的伙伴帮忙解答!

你好同学,可以把

graph

那一句换成

Image(graph.create_png())

就可以显示图像啦