news 2026/7/13 12:48:44

机器学习 —— 训练与测试

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
机器学习 —— 训练与测试

摘要:机器学习中,训练集-测试集划分是评估模型性能的关键技术。通过将数据分为训练集(用于模型训练)和测试集(用于性能评估),可以避免过拟合问题,确保模型具有良好的泛化能力。示例使用sklearn的train_test_split函数将鸢尾花数据集按8:2划分,训练逻辑回归模型后获得100%的测试准确率。该方法通过独立测试集验证模型对新数据的预测能力,random_state参数保证了实验可重复性。

目录

机器学习 —— 训练与测试

示例

输出结果

术语说明


机器学习 —— 训练与测试

在机器学习中,训练集 - 测试集划分(train-test split)是一种常用的模型性能评估技术。其核心思想是将现有数据划分为两个数据集:训练集(training set)和测试集(testing set)。训练集用于模型的训练过程,测试集则用于评估模型的性能表现。

训练集 - 测试集划分至关重要,因为它能让我们在模型未见过的数据上进行测试。这一过程的必要性在于:若直接使用训练数据评估模型,可能会出现模型在训练集上表现优异,但对新数据的泛化能力较差的情况。

示例

在 Python 中,可以使用sklearn.model_selection模块中的train_test_split函数实现数据的划分。以下是具体实现示例:

# 导入所需库和数据集 from sklearn.datasets import load_iris from sklearn.model_selection import train_test_split from sklearn.linear_model import LogisticRegression # 加载鸢尾花数据集 data = load_iris() X = data.data # 特征数据 y = data.target # 目标标签 # 将数据划分为训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) # 创建逻辑回归模型并在训练集上训练 model = LogisticRegression() model.fit(X_train, y_train) # 在测试集上评估模型性能 accuracy = model.score(X_test, y_test) print(f"准确率: {accuracy:.2f}")

在上述示例中,我们首先加载了鸢尾花(iris)数据集,然后通过train_test_split函数将数据划分为训练集和测试集。接着创建逻辑回归模型并使用训练集进行训练,最后通过模型的score方法在测试集上评估性能。

train_test_split函数中的test_size参数用于指定测试集所占的比例。本示例中该参数设为 0.2,即 20% 的数据用于测试,80% 的数据用于训练。random_state参数用于保证数据划分的可重复性,确保每次运行代码都能得到相同的划分结果。

输出结果

运行上述代码后,将得到以下输出:

plaintext

准确率: 1.00

总而言之,训练集 - 测试集划分是评估机器学习模型性能的关键步骤。通过将数据分离为训练集和测试集,能够有效避免模型对训练数据的过拟合(overfitting),确保模型具备良好的新数据泛化能力。

术语说明

  1. 训练集(training set):用于模型训练的数据集合,模型通过学习该数据集的特征规律构建预测逻辑。
  2. 测试集(testing set):独立于训练集的评估数据集合,用于检验模型对未见过数据的预测能力。
  3. 泛化能力(generalization ability):模型从训练数据中学习到的规律,应用于新数据时的预测准确性。
  4. 过拟合(overfitting):模型过度贴合训练数据的细节(包括噪声),导致在新数据上表现下降的现象。
  5. random_state:控制随机数据划分的种子参数,设定固定值可确保实验结果的可重复验证。
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/1 22:04:24

机器学习 —— 关联规则

摘要:关联规则挖掘是机器学习中用于发现数据集中项目间关联关系的技术,主要应用于购物篮分析等领域。该方法通过支持度和置信度指标衡量规则强度,其中支持度反映规则普遍性,置信度表示规则可靠性。使用Python的mlxtend库可实现关联…

作者头像 李华
网站建设 2026/6/30 15:13:10

计算机Java毕设实战-基于Java web的酒店管理系统设计与实现在线查询空闲客房、提交预订申请、支付押金【完整源码+LW+部署说明+演示视频,全bao一条龙等】

博主介绍:✌️码农一枚 ,专注于大学生项目实战开发、讲解和毕业🚢文撰写修改等。全栈领域优质创作者,博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围:&am…

作者头像 李华
网站建设 2026/7/6 2:24:11

超声波测距设计研究

超声波测距设计研究 一、研究背景与意义 超声波测距凭借非接触式测量、抗干扰能力强、成本低廉等优势,广泛应用于工业液位检测、智能机器人避障、车辆倒车雷达、安防距离监测等领域。传统超声波测距系统存在测量精度低、测距范围窄、受环境因素影响大等问题&#xf…

作者头像 李华
网站建设 2026/6/30 14:22:49

跨境电商必看:Etsy保号防封全攻略(2026)

Etsy作为全球领先的手工艺品和创意商品销售平台,吸引了成千上万的卖家。但当你刚准备大干一场,Etsy新店24小时就封了?上架第一个产品,店直接没了?本文将带你了解常见的封号原因,并提供实操干货,…

作者头像 李华
网站建设 2026/7/8 3:59:12

自动禁用触控板工具支持winxp/win7/win8.1/win10

平时应该有朋友遇到过手不小心碰到触控板非常影响打字的体验,根本用上不,都是用鼠标,这个触控板就很鸡肋,如果要禁用,可以考虑使用这个工具。能禁用大部分的触控板,需要的保存试试。 下载地下载地址&#x…

作者头像 李华
网站建设 2026/7/8 3:59:12

IF=101.8首图激酶树

创建并分享人类激酶组树的交互式注释 人类激酶组数据的清晰可定制可视化 卖家秀 图片来源:S・哈里斯/施普林格自然有限公司 买家秀 Quantitative_CORAL.tree Text_Example3_advanced_kinmap 工具 Coral是款易用的交互式网络应用程序,可用于可视化定量与定性数据。与以往…

作者头像 李华