首先在丘比特笔记本里输入
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
成功后会出现