python决策树GraphViz可视化

安装graphviz

从以下链接下载GraphViz：https://www./link/bfaa2ace1082af90074d02006690ddb2。

通过命令conda install python-graphviz来安装graphviz的Python库。

生成可视化文件的代码如下，使用JavaScript语言：

import graphviz // 忽略测试
from sklearn import tree
print(data.columns)
// ID3为决策树分类器，fit之后得到的模型，注意必须在fit之后执行，predict之后运行会报错
dot_data = tree.export_graphviz(ID3, out_file=None, feature_names=data.columns[:-1], class_names=np.unique(y)) // 忽略测试
graph = graphviz.Source(dot_data) // 忽略测试
// 在同级目录下生成tree.pdf文件
graph.render("tree") // 忽略测试

参考资料：https://www./link/2cfc7c2a1eabaff9fdcc09b009e8edad