Python画决策树需要用到Graphviz库,该库的导入大家参照这位博主的这篇文章,已经讲解得很详细,我这里就不赘述了。
python导入Graphviz库
我主要记录一下生成.dot的决策树该如何可视化,比如转化成.png文件。
现在我已经有一个叫titanic.dot的文件,里面是生成的决策树。
将该文件拷贝到下载的Graphviz的bin文件夹中。
以管理员身份运行cmd。
进入Graphviz的bin文件夹目录中,代码如下:
cd D:/graphviz-2.38.1401953780/bin/
cd后面有一个空格,bin文件夹目录根据自己的地址改。
D:
运行命令生成.png图片格式的文件:
dot -Tpng titanic.dot -o titanic.png
或者运行命令生成.pdf格式的文件:
dot –Tpdf titanic.dot –o titanic.pdf