weixin_51211240
2021-11-11 17:17
采纳率: 0%
浏览 32

用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

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

1条回答 默认 最新

相关推荐 更多相似问题