使用随机森林模型在digits数据集上执行分类任务

程序功能

使用随机森林模型对digits数据集进行手写数字分类任务。具体步骤如下:

加载数据:从digits数据集中获取手写数字图片的特征和对应的标签。

划分数据:将数据集分为训练集和测试集,测试集占30%。

训练模型:使用随机森林算法训练分类模型。

进行预测:使用训练好的模型对测试集进行预测。

评估模型:输出模型的分类准确率和详细的分类报告,评估模型的分类效果。

代码

python 复制代码
# 导入所需的库
from sklearn.datasets import load_digits
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score, classification_report

# 加载digits数据集
digits = load_digits()
X = digits.data  # 特征
y = digits.target  # 标签

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# 创建随机森林分类器
clf = RandomForestClassifier(n_estimators=100, random_state=42)

# 训练模型
clf.fit(X_train, y_train)

# 预测测试集
y_pred = clf.predict(X_test)

# 评估模型
accuracy = accuracy_score(y_test, y_pred)
print(f"准确率: {accuracy:.2f}")

# 打印分类报告
print("分类报告:")
print(classification_report(y_test, y_pred, target_names=digits.target_names.astype(str)))
相关推荐
jackzhuoa32 分钟前
java小白闯关记第一天(两个数相加)
java·算法·蓝桥杯·期末
Codeking__1 小时前
链表算法综合——重排链表
网络·算法·链表
minji...2 小时前
数据结构 堆(4)---TOP-K问题
java·数据结构·算法
AI_Keymaker3 小时前
一句话生成3D世界:腾讯开源混元3D模型
算法
Leon_vibs3 小时前
当 think 遇上 tool:深入解析 Agent 的规划之道
算法
旧时光巷3 小时前
【机器学习-2】 | 决策树算法基础/信息熵
算法·决策树·机器学习·id3算法·信息熵·c4.5算法
落了一地秋3 小时前
4.5 优化器中常见的梯度下降算法
人工智能·算法·机器学习
前端伪大叔3 小时前
第 5 篇:策略参数怎么调优?Freqtrade hyperopt 快速入门
算法·代码规范
Code季风3 小时前
深入理解令牌桶算法:实现分布式系统高效限流的秘籍
java·算法·微服务
KyollBM4 小时前
【Luogu】每日一题——Day15. P1144 最短路计数 (记忆化搜索 + 图论 + 最短路)
算法·图论