克制趣头条抄袭专栏推荐正文scikit-learn中决议树的可视化一般需要安装graphviz和python的graphviz插件的安装,下面给出详细的安装步骤:首先我们先进入这个网站的官网:http://www.graphviz.org/(此处已添加圈子卡片,请到今日头条客户端检察)下载好了之后,我们双击一路Next举行安装,安装完成之后我们要设置情况变量:设置完情况变量之后我们需要验证一下是否安装设置乐成,我们在window下令行窗口输入dot -version,如果泛起如下效果就表现安装乐成:自此window安装graphviz就安装乐成了,可是此时python还没有和它发生关系,我们还需要安装python的graphviz插件,这个比力容易我们直接使用下令来安装就ok了。安装python插件graphviz: pip install graphviz安装python插件pydotplus。: pip install pydotplus这样就算安装完成了。
安装完成之后我们来使用一下,看是否课时图像化我们的生成树。from sklearn.datasets import load_irisfrom sklearn import treeimport pydotplusiris = load_iris()#载入鸢尾花的数据集clf = tree.DecisionTreeClassifier()#结构分类树clf = clf.fit(iris.data, iris.target)#拟合数据with open("iris.dot", 'w') as f:#现在可以将模型存入dot文件iris.dot。
f = tree.export_graphviz(clf, out_file=f)dot_data = tree.export_graphviz(clf, out_file=None)#用pydotplus生成iris.pdf。graph = pydotplus.graph_from_dot_data(dot_data)graph.write_pdf("iris.pdf")法式运行之后,会生成两个文件,一个是iris.dot,另外一个是iris.pdf,我们打开pdf就可以看到此时我们结构的生成树。至此就可以说明软件安装乐成了,使用这个软件举行可视化我们的决议树的建设是很是的利便的。
注意:有的时候纵然设置了path,python仍然找不到graphviz,这时,可以在代码内里加入这一行:import sysimport osos.environ["PATH"] += os.pathsep + 'D:graphvizbin/'注意后面的路径是你自己的graphviz的bin目录,这样python就可以找到你的目录了,一般不需要,可是真的要是找不到的话,可以使用这个方法。
本文来源:米乐m6体育登录-www.raidaadon.com