目录

  • 1.前言
  • 2.什么是 one-hot 编码?
  • 3.Python 代码

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.]]