安装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免费学习笔记（深入）”；
参考资料：https://www.php.cn/link/2cfc7c2a1eabaff9fdcc09b009e8edad
以上就是python决策树GraphViz可视化的详细内容，更多请关注php中文网其它相关文章！
python怎么学习？python怎么入门？python在哪学？python怎么学才快？不用担心，这里为大家提供了python速学教程(入门到精通)，有需要的小伙伴保存下载就能学习啦！
广告Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
639
收藏
