深度解析Scipy-2017机器学习教程中的决策树与随机森林
2025-07-10 17:17:17作者:明树来
决策树基础概念
决策树是一种直观且易于理解的机器学习算法,它通过一系列"if-else"规则来做出决策,就像人类做决定的过程一样。决策树的核心思想是通过对数据特征进行二元分割来构建预测模型。
决策树的特点
- 数据预处理简单:能够处理各种类型的数据(连续型和离散型),且对特征缩放不敏感
- 非参数模型:模型复杂度会随着数据量的增加而增加,没有固定的参数数量限制
- 可解释性强:决策规则清晰,容易理解
决策树回归
决策树不仅可以用于分类,也可以用于回归任务。在回归问题中,决策树通过将输入空间划分为多个区域,并在每个区域内预测一个常数值(通常是该区域内目标值的平均值)。
from sklearn.tree import DecisionTreeRegressor
reg = DecisionTreeRegressor(max_depth=5)
reg.fit(X, y)
决策树回归的优缺点
优点:
- 能够捕捉非线性关系
- 对异常值不敏感
- 不需要特征缩放
缺点:
- 容易过拟合(表现为预测结果中的"尖峰")
- 在某些区域可能欠拟合(表现为预测结果中的"平坦"部分)
决策树分类
决策树分类器的工作原理与回归类似,但在每个叶节点上预测的是多数类别而不是平均值。
from sklearn.tree import DecisionTreeClassifier
clf = DecisionTreeClassifier(max_depth=5)
clf.fit(X_train, y_train)
关键参数:max_depth
max_depth参数控制树的深度,直接影响模型的复杂度:
- 深度过小:模型欠拟合,无法捕捉数据中的复杂模式
- 深度过大:模型过拟合,对训练数据中的噪声过于敏感
随机森林
随机森林通过构建多棵决策树并平均它们的预测结果来改善单棵决策树的过拟合问题。每棵树使用:
- 不同的数据子集(有放回抽样)
- 不同的特征子集(无放回抽样)
from sklearn.ensemble import RandomForestClassifier
rf = RandomForestClassifier(n_estimators=200)
rf.fit(X_train, y_train)
随机森林的优势
- 降低方差:通过平均多棵树的预测结果,减少过拟合
- 提高泛化能力:比单棵决策树表现更稳定
- 特征重要性:可以评估每个特征对预测的贡献程度
梯度提升树
梯度提升是另一种集成方法,它通过顺序构建决策树来迭代改进模型。每棵树都试图纠正前一棵树的错误。
from sklearn.ensemble import GradientBoostingClassifier
gbc = GradientBoostingClassifier(n_estimators=100, learning_rate=0.1)
gbc.fit(X_train, y_train)
关键参数
n_estimators:树的数量learning_rate:每棵树对最终结果的贡献程度max_depth:每棵树的深度
模型选择与调优
使用网格搜索交叉验证可以找到最优的模型参数组合:
from sklearn.model_selection import GridSearchCV
parameters = {'max_depth':[5,7,9], 'learning_rate':[0.1,0.01]}
clf_grid = GridSearchCV(gbc, parameters)
clf_grid.fit(X_train, y_train)
特征重要性分析
随机森林和梯度提升树都可以计算特征重要性,这有助于理解哪些特征对预测最有贡献:
importances = rf.feature_importances_
plt.bar(range(len(importances)), importances)
实践建议
- 对于小型数据集,优先尝试梯度提升树
- 对于大型数据集,随机森林通常更高效
- 始终使用交叉验证来评估模型性能
- 关注特征重要性,可以帮助特征工程和模型解释
决策树及其集成方法在实际应用中表现优异,特别是在需要模型可解释性的场景中。通过合理调参和集成,可以获得既准确又稳定的预测模型。
登录后查看全文
热门项目推荐
GLM-5智谱 AI 正式发布 GLM-5,旨在应对复杂系统工程和长时域智能体任务。Jinja00
GLM-5-w4a8GLM-5-w4a8基于混合专家架构,专为复杂系统工程与长周期智能体任务设计。支持单/多节点部署,适配Atlas 800T A3,采用w4a8量化技术,结合vLLM推理优化,高效平衡性能与精度,助力智能应用开发Jinja00
jiuwenclawJiuwenClaw 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。Python0203- QQwen3.5-397B-A17BQwen3.5 实现了重大飞跃,整合了多模态学习、架构效率、强化学习规模以及全球可访问性等方面的突破性进展,旨在为开发者和企业赋予前所未有的能力与效率。Jinja00
AtomGit城市坐标计划AtomGit 城市坐标计划开启!让开源有坐标,让城市有星火。致力于与城市合伙人共同构建并长期运营一个健康、活跃的本地开发者生态。01
awesome-zig一个关于 Zig 优秀库及资源的协作列表。Makefile00
热门内容推荐
项目优选
收起
deepin linux kernel
C
27
12
OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
607
4.05 K
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
69
21
暂无简介
Dart
849
205
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
1.47 K
829
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
12
1
喝着茶写代码!最易用的自托管一站式代码托管平台,包含Git托管,代码审查,团队协作,软件包和CI/CD。
Go
24
0
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
924
772
🎉 基于Spring Boot、Spring Cloud & Alibaba、Vue3 & Vite、Element Plus的分布式前后端分离微服务架构权限管理系统
Vue
235
152
昇腾LLM分布式训练框架
Python
131
157