可视化决策树模型

最新推荐文章于 2024-07-15 20:38:43 发布

炼丹师666

最新推荐文章于 2024-07-15 20:38:43 发布

阅读量728

点赞数 1

分类专栏：机器学习案例算法

本文链接：https://blog.csdn.net/wj1298250240/article/details/103642252

版权

算法同时被 2 个专栏收录

101 篇文章 5 订阅

订阅专栏

机器学习案例

40 篇文章 3 订阅

订阅专栏

可视化决策树模型

创建dot数据

pdf 生成png图片


# 可视化决策树模型
import pydotplus
from sklearn.tree import DecisionTreeClassifier
from sklearn import datasets
from IPython.display import Image
from sklearn import tree

iris = datasets.load_iris()
features = iris.data
target = iris.target

decisiontree = DecisionTreeClassifier(random_state=0)

model = decisiontree.fit(features, target)

# 创建dot数据
dot_data = tree.export_graphviz(decisiontree, out_file=None, feature_names=iris.feature_names, class_names=iris.target_names)
# 画图
graph = pydotplus.graph_from_dot_data(dot_data)
Image(graph.create_png())


# create pdf
#graph.write_pdf("iris.pdf")

# create png PNG 生成png图片
#graph.write_png("iris.png)