安装graphviz
从以下链接下载GraphViz:https://www.php.cn/link/bfaa2ace1082af90074d02006690ddb2。
通过命令conda install python-graphviz来安装graphviz的Python库。
生成可视化文件的代码如下,使用JavaScript语言:
import graphviz // 忽略测试from sklearn import treeprint(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") // 忽略测试

可赞AI
文字一秒可视化,免费AI办公神器
56 查看详情
立即学习“Python免费学习笔记(深入)”;
参考资料:https://www.php.cn/link/2cfc7c2a1eabaff9fdcc09b009e8edad
以上就是python决策树GraphViz可视化的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/512464.html
微信扫一扫
支付宝扫一扫