今天一定要和大家分享以下graphviz!!!
为什么选择它!
它很好,它很棒,它画的图很好!
遇到的bug
我们直接使用pip install可以使用嘛?
是不可以的,但是为什么好多包我们可以直接pip install之后就可以使用了呢?
原因是我们使用的anaconda很厉害,它封装了这些包,我们在创建的虚拟环境中需要使用,直接pip就可以了,十分方便,这也就是我们为什么要使用conda的原因,它对python的编程真的是很友好!
所以,我们一定要先下载graphviz并配置到系统变量
点这里,去官网下载
注意选对应的系统,比如windows-64bit(大家应该都知道,有疑惑的评论区见!!!)
安装到这步的时候直接把添加到环境变量给勾上就行了,这样就不用自己再去配了
你说我不放心,那我们去环境变量里面看看到底有没有!
我下载到了d盘的program files文件,我们可以看到环境变量一般都是到bin文件!!!
配置完后仍然报错:如果
我们通过指令查看是否安装成功
dot -version
如图所示表示已经安装成功了!
那为什么我们还会报错呢!所以,重启是最后的出路了!(之前已经pip install 过了!)
重启成功跑通分类树并画图
代码如下
from sklearn import tree
from sklearn.datasets import load_wine
from sklearn.model_selection import train_test_split
wine=load_wine()
Xtrain,Xtest,Ytrain,Ytest=train_test_split(wine.data,wine.target,test_size=0.3)
clf = tree.DecisionTreeClassifier(criterion="entropy")
clf = clf.fit(Xtrain,Ytrain)
import graphviz
feature_name = ['酒精','苹果酸','灰','灰的碱性','镁','总酚','类黄酮','非黄烷类酚类','花青素','颜色强度','色调','od280/od315稀释葡萄酒','脯氨酸']
dot_data=tree.export_graphviz(clf
,feature_names=feature_name
,class_names=["琴酒","雪梨","贝尔摩德"]
,filled=True
,rounded=True
)
graph=graphviz.Source(dot_data)
graph