1.前言
今天看Colab上的代码的时候突然看到了这个概念,抱着好奇的心态学习了一下,挺基础也很重要的一个概念,5min左右就可以了解
2.什么是 one-hot 编码?
定义:
独热编码即 One-Hot 编码,又称一位有效编码。其方法是使用 N 位状态寄存器来对 N 个状态进行编码,每个状态都有它独立的寄存器位,并且在任意时候,其中只有一位有效。
One-Hot编码是分类变量作为二进制向量的表示。这首先要求将分类值映射到整数值。然后,每个整数值被表示为二进制向量,除了整数的索引之外,它都是零值,它被标记为1。
看不懂么?看不懂很正常(看懂了我觉得不太正常hhh,下面通过举例子很好理解:
这里有三个特征:
性别特征:[‘女’, ‘男’]
国籍特征:[‘中国’, ‘美国’, ‘俄罗斯’, ‘英国’]
年龄特征:[18, 19, 20]
那么,我们现在表示一个20岁中国国籍的男生,那么他的特征为:[‘男’, ‘中国’, 20],我们将特征数字化,即表示为:[1, 0, 2],但是这样的特征放入 Machine Learning 中是不可以的,因为类别之间是无序的,针对这种情况,我们介绍 one-hot 编码:
对于性别特征,有两种取值:女/男,我们用 10(二进制) 表示女,用 01(二进制)表示男
对于国籍特征,有四种取值,我们用 1000(二进制)表示中国,0100(二进制)表示美国,0010(二进制)表示俄罗斯,0001(二进制)表示英国
对于年龄特征,有三种取值,我们用 100(二进制)表示18岁,010(二进制)表示19岁,001(二进制)表示20岁
故对于20岁中国国籍的男生,我们可以表示为:[011000001],用下面这张图更能形象化理解:
3.Python 代码
下述代码在 Jupyter 上运行,没有安装过 Jupyter 的读者可以看博客:最详细的Anaconda Installers 的安装【numpy,jupyter】(图+文)
from sklearn import preprocessing enc = preprocessing.OneHotEncoder() # 调用OneHotEncoder() enc.fit([[0, 2, 2], [1, 1, 4], [1, 0, 1], [0, 1, 3]]) # 提供训练数据:4个数据,3种特征 array = enc.transform([[0,1,3]]).toarray() # 测试,随便输入一个新数据去测试 array # 独热编码结果:[[1., 0., 0., 1., 0., 0., 0., 1., 0.]]