cart算法python实现:从CART算法中学习如何构建有效的决策树

CART(Classification and Regression Tree)算法是一种基于树的机器学习算法,用于分类和回归分析。它使用一种叫做分类和回归树(CART)的决策树结构,通过将数据集分割成多个子集来建立模型。

CART(Classification and Regression Tree)算法是一种基于树的机器学习算法,用于分类和回归分析。它使用一种叫做分类和回归树(CART)的决策树结构,通过将数据集分割成多个子集来建立模型。

CART算法的Python实现如下:

导入必要的库

import numpy as np

from sklearn import datasets

from sklearn.model_selection import train_test_split

from sklearn.tree import DecisionTreeClassifier

加载数据集

dataset = datasets.load_iris()

X = dataset.data

y = dataset.target

将数据集分割成训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=1)

使用CART算法训练模型

clf = DecisionTreeClassifier(criterion='gini')

clf.fit(X_train, y_train)

预测测试集

y_pred = clf.predict(X_test)

计算准确率

accuracy = np.sum(y_pred == y_test)/len(y_test)

print('Accuracy: %.2f' % accuracy)

相关推荐
&&Citrus2 小时前
【CPN学习笔记(二)】Chap2 非分层颜色 Petri 网——从一个简单协议开始读懂 CPN
笔记·学习·php·cpn·petri网
Mr_Xuhhh2 小时前
Java泛型进阶:从基础到高级特性完全指南
开发语言·windows·python
汀、人工智能2 小时前
[特殊字符] 第40课:二叉树最大深度
数据结构·算法·数据库架构·图论·bfs·二叉树最大深度
沉鱼.442 小时前
第十二届题目
java·前端·算法
老天文学家了3 小时前
蓝桥杯备战Python
开发语言·python
大熊背3 小时前
ISP Pipeline中Lv实现方式探究之三--lv计算定点实现
数据结构·算法·自动曝光·lv·isppipeline
小橘子8314 小时前
(学习)Claude Code 源码架构深度解析
学习·程序人生·架构
ID_180079054734 小时前
除了 Python,还有哪些语言可以解析 JSON 数据?
开发语言·python·json
西岸行者4 小时前
BF信号是如何多路合一的
算法
大熊背4 小时前
ISP Pipeline中Lv实现方式探究之一
算法·自动白平衡·自动曝光