决策树算法是风控策略人员必备的进阶能力,今天小编就给大家带来决策树可视化教程
在进行决策树可视化之前,需要有一个必备动作即安装graphviz把代码生成dot文件
windows
https://graphviz.gitlab.io/_pages/Download/windows/graphviz-2.38.msi
下载安装即可
linux
[root@123456 ~]# yum -y install graphviz
[root@123456 ~]# which dot #查看是否安装成功
/usr/bin/dot
代码实现
安装完毕后,就可以随意所欲可视化了,下面以一个例子为例进行简单说明。
import pandas graph.write_pdf(path+ return img