快捷美丽的决策树可视化(无需安装 graphviz 与其他插件)

👆 关注一下~,更多商业数据分析案例等你来撩

本文将专注于决策树的可视化过程,建模原理、结果分析和模型优化则会在未来的博文中呈现。P.S:决策树为客户流失预警项目的必会技能,相关数据源和超详细的代码解析可空降文末~~

前言

以下是笔者见过的对可视化初学者们最友好的概括:数据可视化是数据视觉表现形式的科学技术研究,主要是借助图形化的手段来清晰有效地传达与沟通信息。但这绝不意味着数据可视化必须为实现其功能而去深究枯燥的绘图代码,亦或是为了使图画看上去高端绚丽而显得过于复杂


需求

决策树模型是解释性模型的鼻祖,也是最好解释的模型.最难解释的则是神经网络,所以TA 又被称为“黑盒模型”。所以两者一块学可以更好的比较优缺点。有些实践案例和业务运用上只需要效果好,能较快的响应和被使用就行,那决策树是一个不错的选择;如果领导坚持要求模型可解释,而且一定要在数据库中落地(如能写出 sql,让业务人员一下子就看懂,而且也能获得一些灵感),决策树就是一个很不错的模型。其实用户画像就是用户的特征,决策树也可以用来做用户画像。

决策树建模的结果需要可视化后才能很好的看出分类过程,方法便捷的同时还要兼具一定的美感。搜索引擎以及一些常见的程序员求助社区(Baidu, Google, CSDN, Stackoverflow, 简书,博客园,知乎等)等反馈的最常用的决策树可视化方法的关键词都包含 dot,Graphviz,先来看一下主要代码:


代码也不长,只需要往 sklearn.tree 的 export_graphviz 函数中传入指定参数即可。要想使上述代码成功运行,需安装额外的包还不算什么,但相信 graphviz 的配置过程还是难到了一部分 windows 客户,以下为官方用法的步骤拆解:

使用dot文件进行决策树可视化需要安装一些工具:

第一步:安装graphviz。linux可以用apt-get或者yum的方法安装。如果是windows,就在官网下载msi文件安装。无论是
linux 还是 windows,装完后都要设置环境变量,将 graphviz 的 bin 目录加到 PATH,比如 windows,将
C:/Program Files (x86)/Graphviz2.38/bin/加入了PATH

第二步:安装 python 插件 graphviz:pip install graphviz

第三步:安装 python 插件 pydotplus: pip install pydotplus

配置方面还是有点麻烦的,而且 “ graphviz + dot ” 的这个组合画出的许多图又都是如下这样的,在一定程度上欠缺美感。



所以笔者通过广阅资料和优秀的技术博文,加以实操总结和优化,最终发现 sklearn.tree 里面就已经有了自动实现决策树可视化的函数,只需通过添加不同参数并结合 matplotlib 便可快速绘制出较为精美的决策树,对数据可视化初学者十分友好。


效果实现

篇幅原因,这里只展示核心的可视化过程,简单的建模过程都放在了后台资源,文末回复关键词即可获取。

方法1:只添加最基本的参数(跟没可视化没什么区别,可读性很糟糕)



方法2:通过结合 matplotlib,添加参数来优化可视化结果



注:相关数据源和超详细的代码(python,Jupyter Notebook 版本 + 详细注释)已经整理好,在 “ 数据分析与商业实践 ” 公众号后台回复 “ 决策树可视化 ” 即可获取。

精彩回顾

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,340评论 5 467
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,762评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,329评论 0 329
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,678评论 1 270
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,583评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 47,995评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,493评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,145评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,293评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,250评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,267评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,973评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,556评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,648评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,873评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,257评论 2 345
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,809评论 2 339