0

0

解决LabelEncoder在训练集和测试集上出现“未见标签”错误

DDD

DDD

发布时间:2025-08-25 20:02:20

|

333人浏览过

|

来源于php中文网

原创

解决labelencoder在训练集和测试集上出现“未见标签”错误

本文旨在帮助读者理解并解决在使用LabelEncoder对分类变量进行编码时,遇到的“y contains previously unseen labels”错误。通过详细分析错误原因,并提供正确的编码方法,确保模型在训练集和测试集上的一致性,避免数据泄露。

问题分析

在使用LabelEncoder对分类变量进行编码时,常见的错误是尝试在DataFrame的每一列上单独使用fit_transform方法。这种做法会导致LabelEncoder在每一列上独立地学习标签到数字的映射关系。当测试集中出现训练集中未见过的标签时,或者反过来,就会抛出“y contains previously unseen labels”错误。

例如,假设有一个Education列,包含Bachelor's, Master's, PhD, 和 High School 四个类别。如果在训练集上使用fit_transform,LabelEncoder可能会将它们分别映射到0, 1, 2, 3。如果测试集只包含Master's, PhD, 和 High School,那么当LabelEncoder遇到Bachelor's时,就会报错,因为它没有在训练集中见过这个标签。反之亦然。

正确的编码方法

为了避免这个问题,正确的做法是在整个列上使用fit_transform方法,并且只在训练集上fit一次,然后在训练集和测试集上都使用transform方法。这样可以确保LabelEncoder使用相同的映射关系来编码训练集和测试集。

以下是使用循环的示例代码,展示了如何正确地对多个列进行编码:

蝉妈妈AI
蝉妈妈AI

电商人专属的AI营销助手

下载
from sklearn.preprocessing import LabelEncoder
import pandas as pd

# 假设 cv_df 和 tr_df 是你的训练集和测试集 DataFrame
# encodable_columns 是需要编码的列的列表
encodable_columns = ['Education', 'EmploymentType', 'MaritalStatus', 
                     'HasMortgage', 'HasDependents', 'LoanPurpose', 'HasCoSigner']

# 创建 LabelEncoder 对象
label_encoder = LabelEncoder()

# 在训练集上拟合 LabelEncoder,学习标签到数字的映射
for col in encodable_columns:
    tr_df[col] = label_encoder.fit_transform(tr_df[col])

# 在测试集上使用相同的 LabelEncoder 进行转换
for col in encodable_columns:
    cv_df[col] = label_encoder.transform(cv_df[col])

注意: 在上面的代码中,我们首先创建了一个LabelEncoder对象。然后,我们遍历需要编码的列,并在训练集上使用fit_transform方法来学习标签到数字的映射。最后,我们在测试集上使用transform方法,使用相同的映射关系来编码测试集。

重要提示: 必须先在训练集上fit,然后在训练集和测试集上都使用transform。如果先在测试集上fit,然后再在训练集上transform,可能会导致数据泄露,并且模型在训练集上的表现会很差。

总结

使用LabelEncoder对分类变量进行编码时,务必确保在整个列上使用fit_transform方法,并且只在训练集上fit一次,然后在训练集和测试集上都使用transform方法。这样可以避免“y contains previously unseen labels”错误,并确保模型在训练集和测试集上的一致性。

相关专题

更多
php源码安装教程大全
php源码安装教程大全

本专题整合了php源码安装教程,阅读专题下面的文章了解更多详细内容。

65

2025.12.31

php网站源码教程大全
php网站源码教程大全

本专题整合了php网站源码相关教程,阅读专题下面的文章了解更多详细内容。

43

2025.12.31

视频文件格式
视频文件格式

本专题整合了视频文件格式相关内容,阅读专题下面的文章了解更多详细内容。

35

2025.12.31

不受国内限制的浏览器大全
不受国内限制的浏览器大全

想找真正自由、无限制的上网体验?本合集精选2025年最开放、隐私强、访问无阻的浏览器App,涵盖Tor、Brave、Via、X浏览器、Mullvad等高自由度工具。支持自定义搜索引擎、广告拦截、隐身模式及全球网站无障碍访问,部分更具备防追踪、去谷歌化、双内核切换等高级功能。无论日常浏览、隐私保护还是突破地域限制,总有一款适合你!

41

2025.12.31

出现404解决方法大全
出现404解决方法大全

本专题整合了404错误解决方法大全,阅读专题下面的文章了解更多详细内容。

204

2025.12.31

html5怎么播放视频
html5怎么播放视频

想让网页流畅播放视频?本合集详解HTML5视频播放核心方法!涵盖<video>标签基础用法、多格式兼容(MP4/WebM/OGV)、自定义播放控件、响应式适配及常见浏览器兼容问题解决方案。无需插件,纯前端实现高清视频嵌入,助你快速打造现代化网页视频体验。

9

2025.12.31

关闭win10系统自动更新教程大全
关闭win10系统自动更新教程大全

本专题整合了关闭win10系统自动更新教程大全,阅读专题下面的文章了解更多详细内容。

8

2025.12.31

阻止电脑自动安装软件教程
阻止电脑自动安装软件教程

本专题整合了阻止电脑自动安装软件教程,阅读专题下面的文章了解更多详细教程。

3

2025.12.31

html5怎么使用
html5怎么使用

想快速上手HTML5开发?本合集为你整理最实用的HTML5使用指南!涵盖HTML5基础语法、主流框架(如Bootstrap、Vue、React)集成方法,以及无需安装、直接在线编辑运行的平台推荐(如CodePen、JSFiddle)。无论你是新手还是进阶开发者,都能轻松掌握HTML5网页制作、响应式布局与交互功能开发,零配置开启高效前端编程之旅!

2

2025.12.31

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
10分钟--Midjourney创作自己的漫画
10分钟--Midjourney创作自己的漫画

共1课时 | 0.1万人学习

Midjourney 关键词系列整合
Midjourney 关键词系列整合

共13课时 | 0.9万人学习

AI绘画教程
AI绘画教程

共2课时 | 0.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号