首先在丘比特笔记本里输入
pip install graphviz
显示如下图,表示安装成功。
#引入load_iris from sklearn.datasets import load_iris #引入tree from sklearn import tree #用于画出图形 import graphviz 在window设置graphviz的路径 import os os.environ["PATH"] += os.pathsep + 'E:/Graphviz 2.44.1/bin' iris = load_iris() #创建决策树分类器 clf = tree.DecisionTreeClassifier() #训练 clf = clf.fit(iris.data, iris.target) dot_data = tree.export_graphviz(clf,out_file=None,filled=True,rounded=True,special_characters=True) graph = graphviz.Source(dot_data) graph
成功后会出现
如果运行报错了,可以参考安装graphviz报错:There is no layout engine support for “dot“https://blog.csdn.net/qq_51196770/article/details/120621143



