【Python机器学习系列】一文教你实现决策树模型可视化(案例)

off999 2024-11-26 07:23 29 浏览 0 评论

这是我的第335篇原创文章。

一、引言

决策树是一个有监督分类模型，本质是选择一个最大信息增益的特征值进行输的分割，直到达到结束条件或叶子节点纯度达到阈值。根据分割指标和分割方法，可分为：ID3、C4.5、CART算法。每一种颜色代表一个class，link的宽度表示从一个节点流向另一个节点的items数量。

需要安装第三方库：

pip install pybaobabdt
pip install pygraphviz

二、实现过程

2.1 准备数据

data = pd.read_csv(r'Dataset.csv')
df = pd.DataFrame(data)
print(df.head())

df：

2.2 提取特征变量和目标变量

target = 'target'
features = df.columns.drop(target)
print(data["target"].value_counts()) # 顺便查看一下样本是否平衡

2.3 划分训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(df[features], df[target], test_size=0.2, random_state=0)

2.4 模型构建和训练

model = DecisionTreeClassifier(max_depth=5)
model.fit(X_train, y_train)

2.5 决策树模型可视化

ax = pybaobabdt.drawTree(model, size=10, dpi=300, features=features) #可视化主函数pybaobabdt.drawTree
plt.show()

结果：

作者简介：读研期间发表6篇SCI数据算法相关论文，目前在某研究院从事数据算法相关研究工作，结合自身科研实践经历持续分享关于Python、数据分析、特征工程、机器学习、深度学习、人工智能系列基础知识与案例。关注gzh：数据杂坛，获取数据和源码学习更多内容。

原文链接：

【Python机器学习系列】一文教你实现决策树模型可视化（案例+源码）

python决策树

上一篇：Python决策树
下一篇：通俗易懂的机器学习——根据CART算法使用python构建决策树

【Python机器学习系列】一文教你实现决策树模型可视化(案例)

一、引言

二、实现过程

2.1 准备数据

2.2 提取特征变量和目标变量

2.3 划分训练集和测试集

2.4 模型构建和训练

2.5 决策树模型可视化

相关推荐

取消回复欢迎你发表评论:

抖音上好看的小姐姐，Python给你都下载了

全网最简单易懂!495页Python漫画教程，高清PDF版免费下载

Python 3.14 的 UUIDv6/v7/v8 上新，别再用 uuid4 () 啦!

python入门到脱坑输入与输出—str()函数

飞牛NAS部署TVGate Docker项目，实现内网一键转发、代理、jx

宝塔面板如何添加免费waf防火墙?（宝塔面板开启https）

Python三目运算基础与进阶_python三目运算符判断三个变量

(新版)Python 分布式爬虫与 JS 逆向进阶实战吾爱分享

慕ke 前端工程师2024「完整」

失业程序员复习python笔记——条件与循环

【Python机器学习系列】一文教你实现决策树模型可视化(案例)

一、引言

二、实现过程

2.1 准备数据

2.2 提取特征变量和目标变量

2.3 划分训练集和测试集

2.4 模型构建和训练

2.5 决策树模型可视化

相关推荐

取消回复欢迎 你 发表评论:

抖音上好看的小姐姐，Python给你都下载了

全网最简单易懂!495页Python漫画教程，高清PDF版免费下载

Python 3.14 的 UUIDv6/v7/v8 上新，别再用 uuid4 () 啦!

python入门到脱坑 输入与输出—str()函数

飞牛NAS部署TVGate Docker项目，实现内网一键转发、代理、jx

宝塔面板如何添加免费waf防火墙?（宝塔面板开启https）

Python三目运算基础与进阶_python三目运算符判断三个变量

(新版)Python 分布式爬虫与 JS 逆向进阶实战吾爱分享

慕ke 前端工程师2024「完整」

失业程序员复习python笔记——条件与循环

取消回复欢迎你发表评论:

python入门到脱坑输入与输出—str()函数