0

0

如何通过已训练的 OneHotEncoder 获取指定类别的独热编码向量

聖光之護

聖光之護

发布时间:2026-02-14 15:14:01

|

908人浏览过

|

来源于php中文网

原创

如何通过已训练的 OneHotEncoder 获取指定类别的独热编码向量

本文介绍如何从已拟合的 ColumnTransformer 中提取内置的 OneHotEncoder,并利用其 transform() 方法对新类别值(如 'blue')快速生成对应的独热编码向量,避免重复训练或手动映射。

本文介绍如何从已拟合的 `columntransformer` 中提取内置的 `onehotencoder`,并利用其 `transform()` 方法对新类别值(如 `'blue'`)快速生成对应的独热编码向量,避免重复训练或手动映射。

在使用 sklearn 进行特征工程时,常借助 ColumnTransformer 对数据中特定列(如分类列)应用 OneHotEncoder。但完成拟合后,若需对单个新类别(如 'blue')查询其对应的独热编码(如 [1, 0, 0]),直接调用 ct.transform([['blue']]) 会报错——因为 ColumnTransformer 要求输入维度与原始训练数据一致(本例为 3 列),而 [['blue']] 仅含 1 列。

正确做法是:定位并复用内部已拟合的 OneHotEncoder 实例。ColumnTransformer 将各步骤的变换器以字典形式存储在 named_transformers_ 属性中,可通过名称(如 'cat')或索引安全访问:

fitted_encoder = ct.named_transformers_['cat']

该 OneHotEncoder 实例已学习到类别顺序(categories_[0])、输出特征名(get_feature_names_out()),且支持对任意符合格式的新样本进行编码——只需确保输入为二维数组(即 [[value]] 形式):

# 查询 'blue' 的编码结果
encoded_blue = fitted_encoder.transform([['blue']])
print(encoded_blue)  # 输出: [[1. 0. 0.]]

完整可运行示例如下(注意:sparse_output=False 确保返回稠密数组,便于查看):

Windsurf
Windsurf

Codeium团队打造的AI编程助手

下载
from sklearn.compose import ColumnTransformer
from sklearn.preprocessing import OneHotEncoder
import numpy as np

data = np.array([[1, 'red', 14], 
                 [4, 'blue', 54], 
                 [2, 'green', 241], 
                 [7, 'blue', 25]])

ct = ColumnTransformer(
    transformers=[('cat', OneHotEncoder(sparse_output=False), [1])],
    remainder='passthrough'
)

ct.fit(data)  # 拟合(无需显式调用 fit_transform,此处更清晰)
fitted_encoder = ct.named_transformers_['cat']

# 验证编码逻辑
for category in fitted_encoder.categories_[0]:
    encoded = fitted_encoder.transform([[category]])
    print(f"input '{category}' → encoded as {encoded.flatten()}")

输出:

input 'blue' → encoded as [1. 0. 0.]
input 'green' → encoded as [0. 1. 0.]
input 'red' → encoded as [0. 0. 1.]

关键注意事项

  • 必须使用 [[value]](二维结构),而非 [value] 或 'value',否则 transform() 会因维度不匹配报错;
  • 若 OneHotEncoder 初始化时设置了 handle_unknown='ignore' 或 'infrequent_if_exist',则对未见过的类别(如 'yellow')将返回全零向量(需自行判断是否合理);
  • get_feature_names_out() 返回类似 ['x0_blue', 'x0_green', 'x0_red'] 的数组,明确指示每列对应的实际类别,可用于调试或构建可解释性报告;
  • 此方法完全复用已训练模型,无额外计算开销,适用于推理阶段实时编码或特征解释场景。

综上,通过 ct.named_transformers_['name'] 定位子编码器,再调用其 transform() 方法,即可高效、准确地获取任意已知类别的独热编码向量——这是生产环境中推荐的标准实践。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

23

2026.02.13

微博网页版主页入口与登录指南_官方网页端快速访问方法
微博网页版主页入口与登录指南_官方网页端快速访问方法

本专题系统整理微博网页版官方入口及网页端登录方式,涵盖首页直达地址、账号登录流程与常见访问问题说明,帮助用户快速找到微博官网主页,实现便捷、安全的网页端登录与内容浏览体验。

11

2026.02.13

Flutter跨平台开发与状态管理实战
Flutter跨平台开发与状态管理实战

本专题围绕Flutter框架展开,系统讲解跨平台UI构建原理与状态管理方案。内容涵盖Widget生命周期、路由管理、Provider与Bloc状态管理模式、网络请求封装及性能优化技巧。通过实战项目演示,帮助开发者构建流畅、可维护的跨平台移动应用。

7

2026.02.13

TypeScript工程化开发与Vite构建优化实践
TypeScript工程化开发与Vite构建优化实践

本专题面向前端开发者,深入讲解 TypeScript 类型系统与大型项目结构设计方法,并结合 Vite 构建工具优化前端工程化流程。内容包括模块化设计、类型声明管理、代码分割、热更新原理以及构建性能调优。通过完整项目示例,帮助开发者提升代码可维护性与开发效率。

8

2026.02.13

Redis高可用架构与分布式缓存实战
Redis高可用架构与分布式缓存实战

本专题围绕 Redis 在高并发系统中的应用展开,系统讲解主从复制、哨兵机制、Cluster 集群模式及数据分片原理。内容涵盖缓存穿透与雪崩解决方案、分布式锁实现、热点数据优化及持久化策略。通过真实业务场景演示,帮助开发者构建高可用、可扩展的分布式缓存系统。

3

2026.02.13

c语言 数据类型
c语言 数据类型

本专题整合了c语言数据类型相关内容,阅读专题下面的文章了解更多详细内容。

26

2026.02.12

雨课堂网页版登录入口与使用指南_官方在线教学平台访问方法
雨课堂网页版登录入口与使用指南_官方在线教学平台访问方法

本专题系统整理雨课堂网页版官方入口及在线登录方式,涵盖账号登录流程、官方直连入口及平台访问方法说明,帮助师生用户快速进入雨课堂在线教学平台,实现便捷、高效的课程学习与教学管理体验。

9

2026.02.12

豆包AI网页版入口与智能创作指南_官方在线写作与图片生成使用方法
豆包AI网页版入口与智能创作指南_官方在线写作与图片生成使用方法

本专题汇总豆包AI官方网页版入口及在线使用方式,涵盖智能写作工具、图片生成体验入口和官网登录方法,帮助用户快速直达豆包AI平台,高效完成文本创作与AI生图任务,实现便捷智能创作体验。

181

2026.02.12

PostgreSQL性能优化与索引调优实战
PostgreSQL性能优化与索引调优实战

本专题面向后端开发与数据库工程师,深入讲解 PostgreSQL 查询优化原理与索引机制。内容包括执行计划分析、常见索引类型对比、慢查询优化策略、事务隔离级别以及高并发场景下的性能调优技巧。通过实战案例解析,帮助开发者提升数据库响应速度与系统稳定性。

14

2026.02.12

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Rust 教程
Rust 教程

共28课时 | 5.8万人学习

Git 教程
Git 教程

共21课时 | 3.6万人学习

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

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