美文网首页
决策树-可视化

决策树-可视化

作者: 司马山哥 | 来源:发表于2019-03-28 16:42 被阅读0次

1.Window下安装GraphViz

1.1官网下载

https://graphviz.gitlab.io/_pages/Download/Download_windows.html

graphviz.png

1.2.安装

下载msi然后安装,很简单,一路next, 记住安装路径(下图红色位置,注意格式),后面会设置环境变量


环境变量配置.png

在pycharm中运行pip install graphviz,重启pycharm即可。

2.决策可视化代码

from sklearn.datasets import load_iris
from sklearn import tree
import pydotplus
import graphviz
import os

# 导入路径
os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'

# 导入数据
iris = load_iris()
# 构建模型
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)

# 保存模型
with open("d:\\user\\****\桌面\iris.dot", 'w') as f:
    f = tree.export_graphviz(clf, out_file=f)

# 画图,保存到pdf文件
# 设置图像参数
dot_data = tree.export_graphviz(clf, out_file=None,
                         feature_names=iris.feature_names,
                         class_names=iris.target_names,
                         filled=True, rounded=True,
                         special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
# 保存图像到pdf文件
graph.write_pdf("d:\\user\\***\桌面\iris.pdf")

3.效果

可视化决策树.png

相关文章

  • 决策树

    数据信息 划分数据集 决策树创建 可视化决策树 分析重要信息 决策树的裁剪 可视化结果 其它一些接口

  • python决策树(二叉树、树)的可视化

    问题描述 在我学习机器学习实战-决策树部分,欲可视化决策树结构。最终可视化结果: 解决方案 决策树由嵌套字典组成,...

  • 逼真,特别逼真的决策树可视化

    同学们好,决策树的可视化,我以为之前介绍的方法已经够惊艳了(决策树可视化,被惊艳到了![https://mp.we...

  • 决策树(Decision Tree)算法

    1 理论部分 需要弄清楚几个概念信息熵,决策树,决策树优化, 剪枝 ,决策树可视化 1 信息熵(Entropy 单...

  • 决策树的构建及可视化——帮自己配副隐形眼镜

    本文以一个新的数据集(隐形眼镜数据集)为基础实现构建决策树、决策树的保存与加载、利用决策树分类、决策树的可视化,前...

  • 2018-05-14

    机器学习算法优缺点及其应用领域 决策树 一、 决策树优点 1、决策树易于理解和解释,可以可视化分析,容易提取出规则...

  • 常用机器学习算法优缺点及其应用领域

    机器学习算法优缺点及其应用领域 决策树 一、 决策树优点 1、决策树易于理解和解释,可以可视化分析,容易提取出规...

  • 机器学习算法的特性总结

    决策树 一、 决策树优点1、决策树易于理解和解释,可以可视化分析,容易提取出规则。2、可以同时处理标称型和数值型...

  • 亲手实现决策树(二)

    决策树图像化显示及剪枝 1.决策树可视化 在上一节(亲手实现决策树(一))中,我们利用print_tree进行了文...

  • 【实践篇】决策树的可视化展示

    前面,我们一直在讲决策树的原理,今天呢小鱼就教大家决策树可视化展示的方法,我们来具体形象地看看决策树究竟长什么样子...

网友评论

      本文标题:决策树-可视化

      本文链接:https://www.haomeiwen.com/subject/geccbqtx.html