机器学习:决策树之回归树的原理

目录

前言

一、回归树样例

二、回归树原理

三、原理实现步骤

1.选择特征和分裂点

2.递归分裂

3.生成叶子节点

四、进行预测

总结


前言

决策树回归(Decision Tree Regression)是一种用于预测连续目标变量的回归模型。与分类决策树类似,回归树通过一系列的决策规则对数据进行分割,以最小化预测误差。

一、回归树样例

二、回归树原理

  1. 选择特征和分裂点:在每个节点上,选择一个特征和一个分裂点,将数据集分成两个子集,以最小化某种度量(如均方误差)。
  2. 递归分裂:对每个子集递归地应用相同的分裂规则,直到满足停止条件(如达到最大深度或每个叶子节点包含的样本数过少)。
  3. 生成叶子节点:每个叶子节点输出该节点中样本的平均值作为预测结果。

三、原理实现步骤

1.选择特征和分裂点

  • 先选定一个特征
  • 再选定该特征进行切分
  • 计算选择每个分裂点造成的损失
  • 切分点左边为一部分,右边为一部分
  • 计算两个部分y的方差,再相加得到损失
  • 比较每个切分点的损失,损失最小的即选为根节点

2.递归分裂

  • 对第一个切分点切出来的两个部分,分别再次执行第一步的操作
  • 在两个部分里分别计算每个切分点的损失,找出最少的,当做中间节点

3.生成叶子节点

  • 根据我们选择的深度,判断分裂几次
  • 每个叶子节点的数字都是该节点y的平均值

四、进行预测

  • 只要特征数据在某个范围内,那么得到的预测值都是同一个值

例如:输入的x值为[4,5,6],那么预测值均为6.75

总结

决策树里的参数里树的深度和每个叶子节点的样本数量是两个很重要的参数,调试好这两个参数,能有效地防止决策树过拟合

相关推荐
ZC跨境爬虫8 小时前
跟着 MDN 学 HTML day_9:(信件语义标记)
前端·css·笔记·ui·html
OBiO201312 小时前
Cell | 突破AAV载体容量限制!路中华/姜玉武/刘太安团队开发AAVLINK系统实现大基因递送
笔记
智者知已应修善业12 小时前
【51单片机2个按键控制流水灯运行与暂停】2023-9-6
c++·经验分享·笔记·算法·51单片机
sakiko_13 小时前
UIKit学习笔记5-使用UITableView制作聊天页面
笔记·学习·swift·uikit
Alice-YUE14 小时前
【js高频八股】防抖与节流
开发语言·前端·javascript·笔记·学习·ecmascript
小陈phd15 小时前
TensorRT 入门完全指南(一)——从核心定义到生态工具全解析
人工智能·笔记
是上好佳佳佳呀15 小时前
【前端(十一)】JavaScript 语法基础笔记(多语言对比)
前端·javascript·笔记
handler0115 小时前
Linux 内核剖析:进程优先级、上下文切换与 O(1) 调度算法
linux·运维·c语言·开发语言·c++·笔记·算法
其实防守也摸鱼16 小时前
CTF密码学综合教学指南--第四章
网络·笔记·安全·网络安全·密码学·ctf
05候补工程师18 小时前
【ROS 2 具身智能】Gazebo 仿真避坑指南:从“幽灵机器人”到传感器数据流打通
人工智能·经验分享·笔记·ubuntu·机器人