讲师中心 微信公众号

工具推荐

视频效率加速
Laper
Laper

剧本策划到成片协作

Seko
Seko

AI短剧一键生成成片

AniShort
AniShort

AI短剧协同制作平台

立刻MV
立刻MV

低成本歌曲 MV、视频生成

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

千芳大大_8824

千芳大大_8824

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

|

928人浏览过

|

来源于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 确保返回稠密数组,便于查看):

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工具

更多
创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

相关专题

更多
Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

40

2026.11.06

Postman请求头Headers怎么设置
Postman请求头Headers怎么设置

围绕Postman HTTP请求头配置展开,讲解Headers选项的具体使用方式,包括Key和Value参数填写、请求头信息添加、常见Header字段设置及请求头修改方法,帮助用户解决接口调试过程中遇到的请求头配置问题。

20

2026.11.06

Postman接口请求发送教程
Postman接口请求发送教程

详细介绍Postman接口请求的创建与发送方法,涵盖GET和POST请求、URL配置、请求参数、Headers请求头、Body请求体设置以及响应结果查看,帮助用户掌握Postman接口请求的完整操作流程。

20

2026.11.06

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

40

2026.11.05

DBeaver安全防护教程大全
DBeaver安全防护教程大全

DBeaver安全防护专题,围绕数据库连接安全、账号密码保护和访问权限管理展开,整理SSL/TLS加密连接、SSH隧道、数据库凭据保存及证书验证等配置方法,同时介绍敏感信息保护、远程连接风险和高权限账号使用等注意事项,帮助用户提高数据库日常管理的安全性。

40

2026.11.05

DBeaver连接创建与配置教程
DBeaver连接创建与配置教程

本专题聚焦DBeaver数据库连接配置与管理,覆盖从新建连接、选择数据库类型到填写连接参数和验证连接的完整流程。针对MySQL、PostgreSQL、SQL Server等常见数据库,整理相应的连接设置方法,并提供驱动缺失、认证失败和连接异常等问题的排查方向。

40

2026.11.04

XORM框架结构体映射与表结构同步
XORM框架结构体映射与表结构同步

本专题围绕XORM框架的结构体映射展开,说明表名与字段的命名策略、xorm标签常用属性、主键自增与时间字段自动填充,并演示Sync2自动建表、增量字段升级、索引声明与反向生成结构体的完整流程。

40

2026.11.04

XORM框架安装配置与数据库连接
XORM框架安装配置与数据库连接

本专题聚焦XORM框架的配置与连接环节,拆解MySQL、PostgreSQL、SQLite、SQL Server的DSN写法与驱动选择,讲清连接池参数、多数据源拆分、读写分离与超时重连的处理方式,并补充容器与兼容数据库的适配经验。

40

2026.11.04

XORM框架安装配置与数据库连接
XORM框架安装配置与数据库连接

本专题聚焦XORM框架的配置与连接环节,拆解MySQL、PostgreSQL、SQLite、SQL Server的DSN写法与驱动选择,讲清连接池参数、多数据源拆分、读写分离与超时重连的处理方式,并补充容器与兼容数据库的适配经验。

0

2026.11.04

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
LLVM Programmer’s Manual
LLVM Programmer’s Manual

共0课时 | 0人学习

Conan 包创建教程
Conan 包创建教程

共0课时 | 0人学习

Visual Studio 入门指南
Visual Studio 入门指南

共0课时 | 0人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn