python 决策树可视化

最新推荐文章于 2024-08-01 08:51:58 发布

goongoup

最新推荐文章于 2024-08-01 08:51:58 发布

阅读量239

点赞数

本文链接：https://blog.csdn.net/goongoup/article/details/110426777

版权

对于 CART 回归树的可视化，可以先在电脑上安装 graphviz；然后 pip install graphviz，这是安装python的库，需要依赖前面安装的 graphviz。可视化代码如下：

from sklearn.tree import export_graphviz
import graphviz

# 参数是回归树模型名称，不输出文件。
dot_data = export_graphviz(dtr, out_file=None)
graph = graphviz.Source(dot_data)
# render 方法会在同级目录下生成 Boston PDF文件，内容就是回