0

0

使用 Category Encoders 编码未见数据:避免目标列缺失问题

聖光之護

聖光之護

发布时间:2025-08-01 17:08:01

|

330人浏览过

|

来源于php中文网

原创

使用 category encoders 编码未见数据:避免目标列缺失问题

本文旨在解决在使用 Category Encoders 库时,如何对不包含目标列的未见数据(如测试集)进行编码的问题。重点在于训练编码器时明确排除目标列,从而确保在转换未见数据时列数匹配,避免因列数不一致导致的错误。通过示例代码演示了正确的编码流程,帮助读者理解并应用该方法。

在使用 category_encoders 库进行特征编码时,经常会遇到训练集包含目标列,而测试集(或其他未见数据)不包含目标列的情况。直接使用在训练集上训练的编码器转换测试集,可能会因为列数不匹配而报错。 本文将介绍如何正确地处理这种情况,确保模型能够顺利地对未见数据进行预测。

核心思路:在训练编码器时排除目标列

问题的关键在于,在训练编码器时,要明确告知编码器哪些列是特征列,哪些列是目标列。通常,TargetEncoder 等编码器会自动识别目标列,但当我们需要对不包含目标列的数据集进行编码时,就需要在训练阶段显式地排除目标列。

示例代码

以下是一个完整的示例,展示了如何使用 TargetEncoder 对训练集和测试集进行编码,其中测试集不包含目标列:

DreamGen
DreamGen

一个AI驱动的角色扮演和故事写作的平台

下载
import category_encoders as ce
import pandas as pd

# 假设你已经读取了训练集和测试集
training_data = pd.read_csv("train.csv")
test_data = pd.read_csv("test.csv")

# 将目标列从训练集中分离出来
X_train = training_data.drop("target_column", axis=1)
y_train = training_data["target_column"]

# 初始化 TargetEncoder,并指定要编码的列
# 这里使用 X_train.columns,确保只编码特征列
encoder = ce.TargetEncoder(cols=X_train.columns)

# 在训练集上训练编码器
encoded_train = encoder.fit_transform(X_train, y_train)

# 使用训练好的编码器转换测试集
encoded_test = encoder.transform(test_data)

# encoded_train 和 encoded_test 现在包含了编码后的数据
print("Encoded Training Data:")
print(encoded_train.head())
print("\nEncoded Test Data:")
print(encoded_test.head())

代码解释:

  1. 数据准备: 首先,使用 pandas 读取训练集和测试集。 关键一步是使用 training_data.drop("target_column", axis=1) 将目标列从训练集中移除,并将目标列单独存储在 y_train 中。
  2. 编码器初始化: ce.TargetEncoder(cols=X_train.columns) 初始化 TargetEncoder,并使用 X_train.columns 指定要编码的列。 这样可以确保只对特征列进行编码,避免将目标列包含在内。
  3. 训练和转换: encoder.fit_transform(X_train, y_train) 在训练集上训练编码器,并同时进行转换。 encoder.transform(test_data) 使用训练好的编码器转换测试集。 由于编码器只对特征列进行了训练,因此可以顺利地对不包含目标列的测试集进行编码。

注意事项:

  • 目标列名称: 确保代码中的 "target_column" 与实际目标列的名称一致。
  • 列的顺序: 训练集和测试集的列顺序应该保持一致,除了目标列之外。 否则,编码器可能会将错误的列进行编码。
  • 缺失值处理: 如果数据中存在缺失值,需要在编码之前进行处理,例如使用 fillna() 填充缺失值。
  • 其他编码器: 以上方法也适用于其他 category_encoders 中的编码器,例如 OneHotEncoder、OrdinalEncoder 等。 只需要在初始化编码器时,指定要编码的列即可。

总结

通过在训练编码器时明确排除目标列,可以有效地解决对不包含目标列的未见数据进行编码的问题。 这种方法可以确保编码器只对特征列进行训练,从而避免因列数不匹配而导致的错误,使得模型能够顺利地对未见数据进行预测。 记住,数据准备阶段的列选择至关重要,它直接影响编码器的训练和转换结果。

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

54

2025.12.04

c++ 根号
c++ 根号

本专题整合了c++根号相关教程,阅读专题下面的文章了解更多详细内容。

57

2026.01.23

c++空格相关教程合集
c++空格相关教程合集

本专题整合了c++空格相关教程,阅读专题下面的文章了解更多详细内容。

57

2026.01.23

yy漫画官方登录入口地址合集
yy漫画官方登录入口地址合集

本专题整合了yy漫画入口相关合集,阅读专题下面的文章了解更多详细内容。

236

2026.01.23

漫蛙最新入口地址汇总2026
漫蛙最新入口地址汇总2026

本专题整合了漫蛙最新入口地址大全,阅读专题下面的文章了解更多详细内容。

393

2026.01.23

C++ 高级模板编程与元编程
C++ 高级模板编程与元编程

本专题深入讲解 C++ 中的高级模板编程与元编程技术,涵盖模板特化、SFINAE、模板递归、类型萃取、编译时常量与计算、C++17 的折叠表达式与变长模板参数等。通过多个实际示例,帮助开发者掌握 如何利用 C++ 模板机制编写高效、可扩展的通用代码,并提升代码的灵活性与性能。

17

2026.01.23

php远程文件教程合集
php远程文件教程合集

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

103

2026.01.22

PHP后端开发相关内容汇总
PHP后端开发相关内容汇总

本专题整合了PHP后端开发相关内容,阅读专题下面的文章了解更多详细内容。

73

2026.01.22

php会话教程合集
php会话教程合集

本专题整合了php会话教程相关合集,阅读专题下面的文章了解更多详细内容。

81

2026.01.22

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
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号