python决策树GraphViz可视化-小浪学习网

安装graphviz

从以下链接下载GraphViz：https://www.php.cn/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") // 忽略测试

python决策树GraphViz可视化

立即学习“Python免费学习笔记（深入）”；

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

以上就是

文章版权归作者所有，未经允许请勿转载。

THE END