文档章节

python决策树可视化

bakacx
 bakacx
发布于 2017/08/12 11:52
字数 153
阅读 74
收藏 0

# %%
# 决策树
IMnum = 120
# InitialMonthnum初始训练数据为6年,72个月
train_data = dataset.ix[dtlist[:IMnum],:]
train_X = np.array(train_data[fctname])
train_Y = np.array(train_data['label'])

from sklearn import tree
import matplotlib.pyplot as plt

clf = tree.DecisionTreeClassifier(max_leaf_nodes=6)
clf = clf.fit(train_X, train_Y)

test_data = dataset.ix[dtlist[IMnum:],:]
test_X = np.array(test_data[fctname])
test_Y = np.array(test_data['label'])
clf.score(test_X, test_Y)
# clf.decision_path(train_X)


from sklearn.tree import export_graphviz
export_graphviz(
        clf,
        out_file="tree.dot",
        feature_names=fctname,
        # class_names='label',
        rounded=True,
        filled=True
    )

进入windows命令行界面,cd 切换到tree.dot所在的路径,执行

dot -Tpng tree.dot -o tree.png

 

graphviz安装及使用:

http://blog.csdn.net/lanchunhui/article/details/49472949#%E4%B8%8B%E8%BD%BD%E5%AE%89%E8%A3%85%E9%85%8D%E7%BD%AE%E7%8E%AF%E5%A2%83%E5%8F%98%E9%87%8F

 

 

© 著作权归作者所有

bakacx
粉丝 0
博文 5
码字总数 2573
作品 0
私信 提问
scikit-learn之决策树可视化

  平时我们在用机器学习建模时,往往只是用建模去分析数据,得到结论。但有时,我们也需要一些可视化的东西,比如决策树可视化等。   在Python的机器学习库scikit-learn中,tree类中的e...

但盼风雨来_jc
2018/05/31
0
0
Machine Learning Mastery 博客文章翻译:XGBoost

请您勇敢地去翻译和改进翻译。虽然我们追求卓越,但我们并不要求您做到十全十美,因此请不要担心因为翻译上犯错——在大部分情况下,我们的服务器已经记录所有的翻译,因此您不必担心会因为您...

ApacheCN_飞龙
03/26
0
0
决策树ID3原理及R语言python代码实现(西瓜书)

决策树ID3原理及R语言python代码实现(西瓜书) 摘要: 决策树是机器学习中一种非常常见的分类与回归方法,可以认为是if-else结构的规则。分类决策树是由节点和有向边组成的树形结构,节点表...

知然
08/19
0
0
网贷环绕四周,许多人深陷其中,Python揭露网贷不为人知的一面!

网贷 一些网络贷款平台面向在校大学生开展的贷款业务。据调查,校园消费贷款平台的风控措施差别较大,个别平台存在学生身份被冒用的风险。此外,部分为学生提供现金借款的平台难以控制借款流...

Python新世界
2018/07/17
0
0
统计学习方法第五章:决策树(decision tree),ID3算法,C4.5算法及python实现

统计学习方法第二章:感知机(perceptron)算法及python实现 统计学习方法第三章:k近邻法(k-NN),kd树及python实现 统计学习方法第四章:朴素贝叶斯法(naive Bayes),贝叶斯估计及python实现 ...

无限大的饿
02/17
0
0

没有更多内容

加载失败,请刷新页面

加载更多

从零基础到拿到网易Java实习offer,我做对了哪些事

作为一个非科班小白,我在读研期间基本是自学Java,从一开始几乎零基础,只有一点点数据结构和Java方面的基础,到最终获得网易游戏的Java实习offer,我大概用了半年左右的时间。本文将会讲到...

Java技术江湖
昨天
5
0
程序性能checklist

程序性能checklist

Moks角木
昨天
7
0
VUE 计算属性

本文转载于:专业的前端网站▶VUE 计算属性 1、示例代码 <!DOCTYPE html><html lang="zh"> <head> <meta charset="UTF-8" /> <title>vue示例</title> </hea......

前端老手
昨天
6
0
快速搭建LNMT平台和环境部署 Tomcat详解

Tomcat部署的基本概念 1. CATALINA_HOME与CATALINA_BASE分别指什么?     CATALINA_HOME指的是Tomcat的安装目录     bin:\\Tomcat一些脚本存放目录,比如启动脚本startup.bat/start...

网络小虾米
昨天
7
0
float浮动

float浮动 float浮动概念及原理: 文档流:文档流是文档中可显示对象在排列时所占用的位置。 加浮动的元素,会脱离文档流,会沿父容器靠左或靠右排列,如果之前已经有浮动的元素,会挨着浮动...

studywin
昨天
8
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部