sklearn可视化GraphViz报错解决方法

      在使用sklearn做了数模型后,比如决策树等模型,需要把模型可视化的时候可以选择通过pydotplus包来可视化,初次运行会遇到如下报错:

pydotplus.graphviz.InvocationException: GraphViz's executables not found

本次运行环境是python3.6+ win10 +pycharm

运行部分代码如下:

import pydotplus
decision_tree=tree.DecisionTreeClassifier(criterion='entropy')
tre.fit(x_train_std,y_train)
y_pre=tre.predict(x_test_std)
#visualize the decision tree
dot_data=tree.export_graphviz(decision_tree,out_file=None)
graph=pydotplus.graph_from_dot_data(dot_data)
graph.write_pdf('D:\\model\\decisiontree.pdf')

后面采用的解决方法如下:

在GraphGiz官网下载适合的版本,由于使用的windows,于是下载了windows版本的,

附下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html

一般下载我会直接选择稳定版本下载,所以这里也是下载的稳定版本。

下载完后正常安装好后将安装路径添加到系统环境变量中,如将此路径添加进去,因为我选择的是默认安装,所以装在了C盘

 C:\Program Files (x86)\Graphviz2.38\bin

环境变量添加完后,再重启Pycharm,然后运行就会正确输出pdf文件啦。

猜你喜欢

转载自blog.csdn.net/qq_27575895/article/details/80915466