CART算法原理及Python实践

news2024/9/21 2:36:22

一、CART算法原理

CART(Classification And Regression Trees)算法是一种用于分类和回归任务的决策树学习技术。它采用贪心策略递归地划分数据集,以构建一棵二叉决策树。CART算法的原理可以概括为以下几个关键步骤:

1. 特征选择与数据划分

特征选择:CART算法在每次划分时,会选择最优的特征及其对应的划分点(对于连续特征)或划分值(对于离散特征)。对于分类任务,通常使用基尼指数(Gini Index)作为划分标准;对于回归任务,则使用均方误差(MSE)作为划分标准。基尼指数越小或均方误差越小,表示划分后的数据子集越纯或越接近真实值。

数据划分:根据选定的最优特征和划分点/值,将数据集划分为两个子集。这两个子集将作为新生成的子节点的训练数据集。

2. 递归构建决策树

从根节点开始,CART算法递归地对每个节点进行上述的特征选择和数据划分操作,直到满足停止条件。常见的停止条件包括:节点中的样本个数小于预定的阈值、节点样本的基尼指数小于预定的阈值(分类树)、没有更多特征可以选择等。

递归过程中,每个非叶子节点都会生成两个子节点,从而构建出一棵二叉决策树。

3. 剪枝处理

为了避免过拟合,CART算法会对生成的决策树进行剪枝处理。剪枝的目的是简化决策树的复杂度,提高其对未知数据的预测能力。

CART算法通常采用后剪枝技术,即首先生成一棵完全生长的决策树,然后从树的底端开始,逐步剪去一些子树,直到满足某个停止条件(如损失函数最小)。剪枝过程中,会计算剪枝前后的损失函数变化量,选择使得损失函数最小的剪枝策略。

4. 决策树的应用

构建完成的CART决策树可以用于分类或回归任务。对于分类任务,决策树会根据输入的特征值,沿着决策树的路径进行遍历,直到达到一个叶子节点,该叶子节点对应的类别即为预测结果。

对于回归任务,决策树同样会根据输入的特征值进行遍历,但叶子节点对应的是一个具体的数值预测结果。

总的来说,CART算法通过递归地构建二叉决策树,并结合剪枝技术来提高模型的泛化能力,是一种非常有效且广泛应用的机器学习算法。

二、CART算法的Python实践

在Python中,使用CART算法的一个非常方便的方式是通过scikit-learn库,它提供了DecisionTreeClassifier(用于分类)和DecisionTreeRegressor(用于回归)两个类,这两个类都实现了CART算法。下面我将给出这两个类的简单使用示例。

1、分类任务(使用DecisionTreeClassifier)

首先,你需要安装scikit-learn库(如果你还没有安装的话):

pip install scikit-learn

然后,你可以使用以下代码进行CART分类树的实践:

from sklearn.datasets import load_iris

from sklearn.model_selection import train_test_split

from sklearn.tree import DecisionTreeClassifier

from sklearn import tree

import matplotlib.pyplot as plt

# 加载数据集

iris = load_iris()

X = iris.data

y = iris.target

# 划分训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# 创建决策树分类器实例

clf = DecisionTreeClassifier(criterion='gini', random_state=42)  # 使用基尼指数作为划分标准

# 训练模型

clf.fit(X_train, y_train)

# 预测测试集

y_pred = clf.predict(X_test)

# 评估模型(这里只是简单示例,你可以使用更复杂的评估方法)

accuracy = clf.score(X_test, y_test)

print(f'Accuracy: {accuracy}')

# 可视化决策树(可选)

plt.figure(figsize=(20,10))

tree.plot_tree(clf, filled=True, feature_names=iris.feature_names, class_names=iris.target_names)

plt.show()

2、回归任务(使用DecisionTreeRegressor)

对于回归任务,你可以使用DecisionTreeRegressor类,它的使用方式与DecisionTreeClassifier非常相似,只是目标变量y是连续的而不是离散的。

这里是一个简单的回归任务示例,但请注意,scikit-learn并没有内置用于回归任务的标准数据集,所以我们这里只是演示如何构建和训练模型:

import numpy as np

from sklearn.model_selection import train_test_split

from sklearn.tree import DecisionTreeRegressor

# 假设我们有以下简单的回归数据集(实际使用中,你需要从真实数据源加载数据)

X = np.array([[1], [2], [3], [4], [5], [6], [7], [8], [9], [10]])

y = np.array([2.7, 3.2, 3.8, 5.1, 5.9, 7.1, 7.9, 8.8, 9.2, 10.1])

# 划分训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# 创建决策树回归器实例

regressor = DecisionTreeRegressor(random_state=42)

# 训练模型

regressor.fit(X_train, y_train)

# 预测测试集

y_pred = regressor.predict(X_test)

# 评估模型(这里你可以使用MSE、RMSE等回归评估指标)

from sklearn.metrics import mean_squared_error

mse = mean_squared_error(y_test, y_pred)

print(f'MSE: {mse}')

# 注意:由于决策树回归器的可视化比较复杂,这里不展示可视化代码。

# 如果你需要可视化决策树,可以考虑使用`export_graphviz`函数将树导出为DOT格式,然后用Graphviz软件查看。在上面的示例中,我们使用了scikit-learn提供的数据集(对于分类)和自定义的简单数据集(对于回归)来演示如何使用CART算法进行分类和回归任务。在实际应用中,你需要从真实的数据源中加载数据,并进行适当的数据预处理和特征工程。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2096691.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

如何在 CentOS 6 上安装 Nagios

前些天发现了一个巨牛的人工智能学习网站,通俗易懂,风趣幽默,忍不住分享一下给大家。点击跳转到网站。 Status: 已弃用 本文涵盖的 CentOS 版本已不再受支持。如果您目前正在运行 CentOS 6 服务器,我们强烈建议升级或迁移到受支持…

什么是单片机?为什么要学习单片机?

实现目标 1、熟悉单片机定义、特点、应用场景、发展历史等; 2、理解为什么要学习单片机?怎样学习单片机? 一、单片机是什么? 1、定义 单片机是集成在一块(单)芯片上的微型计算机。平时我们把 MCU&#x…

E. Sheep Eat Wolves

https://codeforces.com/gym/104869/problem/E 赛时队友想贪心,贪不了一点,我想了数学办法每次都送固定的发现送过去就不满足了 赛后补,暴力做O(n4) 至少要几次才能把安全所有羊送到对岸去 考虑最短路,bfs,用数组存下所有状态 …

nerf论文代码理解

近年来,生成式AI(AGI)快速发展,各类生成式模型层出不群,但我更觉得具有物理意义的生成式AI将是未来革命性技术。因此也在抽空看看关于三维重建的知识,这篇文章就是记录我对nerf的理解。 一、论文理解 首先,需要知道ne…

车载冰箱的介绍

车载冰箱产业链的上游环节主要涉及制冷模块的研发与制造,这一环节根据所采用的制冷技术不同而展现出各自的特点。压缩机制冷模块和半 导体制冷模块是车载冰箱制冷系统的核心组成部分,它们在设计、材料选择和制造工艺上各有千秋。 车载冰箱的原理图 车载…

HTB-Three(云服务)

前言 各位师傅大家好,我是qmx_07,今天给大家讲解Three靶机 渗透过程 信息搜集 服务器开启了22 SSH 和 80 HTTP服务端口 扫描目录 服务器只有一个image图片文件,ssh 需要账号名密码 绑定域名 进行子域名扫描探测 疑似域名,尝…

中秋之美——html5+css+js制作中秋网页

中秋之美——html5cssjs制作中秋网页 一、前言二、功能展示三、系统实现四、其它五、源码下载 一、前言 八月十五,秋已过半,是为中秋。 “但愿人长久,千里共婵娟”,中秋时节,气温已凉未寒,天高气爽&#x…

支付平台构建支付接口供整个公司调用—支付代理商

一、支付平台一般需要对外开放用于满足一体化平台调用 支付平台提供支付接口具有以下必要性: 1. 拓展业务范围:使各类商家和服务提供商能够接入支付服务,从而扩大支付平台的用户群体和业务覆盖范围。 2. 提升用户体验:为不同的应…

音视频-图像篇(YUV和RGB)

文章目录 一、图像基础概念二、YUV与RGB1.YUV分类方式2.YUV“空间-间”的数据划分1)UV按照“空间-间”的划分方式,分为YUV444、YUV422、YUV4202)YUV“空间-内”的数据划分 3.RGB 三、比较JPG、PNG、GIF、BMP图片格式 一、图像基础概念 像素&…

Python OpenCV 入门指南

引言 OpenCV(Open Source Computer Vision Library)是一个开源的计算机视觉库,它包含了大量的用于实时图像分析和视频处理的功能。Python 作为一种流行的编程语言,以其易用性和灵活性成为了进行快速原型设计和开发的理想选择。结…

【python2C】排序算法

题&#xff1a;逆序对&#xff08;NXD&#xff09; 对于给定的一段正整数序列a&#xff0c;逆序对就是序列中 a[i]​>a[j]​ 且 i<j 的有序对。 输入格式 第一行&#xff0c;一个正整数 n&#xff0c;表示序列中有 n个数&#xff0c;n<5e5 第二行&#xff0c; n 个正整…

学习之git

github 创建远程仓库 代码推送 Push 代码拉取 Pull 代码克隆 Clone SSH免密登录 Idea集成GitHubGitee码云 码云创建远程仓库 Idea集成Gitee码云 码云连接Github进行代码的复制和迁移GitLab gitlab服务器的搭建和部署 Idea集成GitLabgit概述 一切皆本地 版本控制工具 集中…

读软件开发安全之道:概念、设计与实施15安全测试

1. 安全测试 1.1. 测试是开发可靠、安全代码中的关键一环 1.2. 测试安全漏洞的目的是主动检测 1.3. 模糊测试是一种强大的补充技术&#xff0c;可以帮助我们找到更深层次的问题 1.4. 针对当前漏洞创建的安全回归测试&#xff0c;目的是确保我们不会再犯相同的错误 1.5. 大…

LabVIEW焊缝视觉识别系统

随着自动化技术的发展&#xff0c;焊接工艺也在向智能化和自动化转型。介绍了一种基于LabVIEW开发的自动化焊接机器人视觉识别系统&#xff0c;用于提高焊接质量和效率&#xff0c;特别适用于复杂或危险环境下的操作。 项目背景 在传统焊接工艺中&#xff0c;焊接质量和效率受…

linux 文件管理命令

Ctrl h, 在对应的文件夹路径下显示文件内容 2.“~”、“/”、“./”分别代表什么&#xff1f; https://blog.csdn.net/shuaiqidexiaojiejie/article/details/132563658 2.1&#xff0c;“~”表示主目录&#xff0c;也就是当前登录用户的用户目录 cd ~, 表示返回到Home目录 2.2…

使用 GZCTF 结合 GitHub 仓库搭建独立容器与动态 Flag 的 CTF 靶场以及基于 Docker 的 Web 出题与部署

写在前面 关于 CTF 靶场的搭建&#xff08;使用 CTFd 或者 H1ve&#xff09;以及 AWD 攻防平台的搭建&#xff0c;勇师傅在前面博客已经详细写过&#xff0c;可以参考我的《网站搭建》专栏&#xff0c;前段时间玩那个 BaseCTF&#xff0c;发现它的界面看着挺不错的&#xff0c…

esp8266根据httpserver状态,调用网络唤醒,实现一键开机

esp8266根据httpserver状态&#xff0c;调用网络唤醒&#xff0c;实现一键开机 一.开发板程序二. 服务端三.服务端状态变更 一.开发板程序 #include <ESP8266WiFi.h> #include <ESP8266HTTPClient.h> #include <WiFiUdp.h> #include <ArduinoJson.h>/…

C语言深入理解指针1

1. 内存和地址 1.1内存 内存空间如何高效管理?把内存划分一个个的内存单元&#xff0c;每个内存单元的大小取一个字节一个比特位可以存储一个2进制的位1或者0&#xff0c;那么一个内存单元可以放8个比特位&#xff0c;每个内存单元都有一个编号&#xff0c;有了内存单元的编…

Golang | Leetcode Golang题解之第389题找不同

题目&#xff1a; 题解&#xff1a; func findTheDifference(s, t string) (diff byte) {for i : range s {diff ^ s[i] ^ t[i]}return diff ^ t[len(t)-1] }

自然美景短视频素材哪里找?9个美景视频素材库网站分享

你是否曾被窗外的落日余晖震撼&#xff0c;或是被雄伟的山脉和蔚蓝的大海深深吸引&#xff1f;想要分享这世界的美好&#xff0c;却苦于没有合适的视频素材&#xff1f;今天&#xff0c;就让我带你探索那些充满自然韵味的素材宝库&#xff0c;帮你捕捉和分享自然的美丽&#xf…