标签的独热编码——LabelBinarizer()

[这是一个简单的备忘录]
LabelBinarizer()是sklearn.preprocessing中的一个函数,通过这个函数可以实现机器学习中国对图像标签的独热编码
独热编码是一种二进制编码,通俗的讲独热编码主要满足以下几个条件:

  1. 图像有n类,则编码长度为n
  2. 对第i类编码,只有第i位置1,其余为0 ( 1 ≤ i ≤ n 1 \leq i \leq n 1in)

在使用LabelBinarizer()时,首先要申明一个变量作为其载体
(具体名词我不知道) ,再分别调用fit与transform函数实现。
for example:

import glob
from sklearn.preprocessing import LabelBinarizer
DATA_DIR = 'D:/Data/'
# 总共三个类别,每个类别下有十张图片
images = glob.glob(DATA_DIR + 'fruits_photos/*/*.jpg')
images = [x.replace('\\', '/') for x in images]
labels = [x.split('/')[3] for x in images]

unique_labels = set(labels)

# 将标签转换二进制格式
encoder = LabelBinarizer()
print(encoder)
encoder.fit(labels)
y = encoder.transform(labels).astype(float)
print(y)

输出结果为:

LabelBinarizer()
[[1. 0. 0.]
[1. 0. 0.]
[1. 0. 0.]
[1. 0. 0.]
[1. 0. 0.]
[1. 0. 0.]
[1. 0. 0.]
[1. 0. 0.]
[1. 0. 0.]
[1. 0. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 1. 0.]
[0. 0. 1.]
[0. 0. 1.]
[0. 0. 1.]
[0. 0. 1.]
[0. 0. 1.]
[0. 0. 1.]
[0. 0. 1.]
[0. 0. 1.]
[0. 0. 1.]
[0. 0. 1.]]

在sklearn中进行独热编码有几种方式。一种常用的方法是先使用LabelEncoder将分类特征进行数字编码,然后再使用OneHotEncoder将数字编码转化为独热编码。另外一种方法是直接使用LabelBinarizer对文本进行独热编码。还有一种方法是使用DictVectorizer。 首先,我们可以使用LabelEncoder将分类特征进行数字编码。通过以下代码可以实现: from sklearn.preprocessing import LabelEncoder encoder = LabelEncoder() category = encoder.fit_transform(data['颜色']) print("颜色编码:",category) print("编码对应的颜色:",encoder.classes_) 其中,data['颜色']表示待编码的分类特征列,category是编码后的结果,encoder.classes_表示编码对应的原始分类特征。 接着,我们可以使用OneHotEncoder将数字编码转化为独热编码。可以使用以下代码实现: from sklearn.preprocessing import OneHotEncoder encoder = OneHotEncoder() onehot = encoder.fit_transform(category.reshape(-1, 1)).toarray() print("独热编码结果:", onehot) 其中,category.reshape(-1, 1)是将数字编码转化为二维数组,toarray()用于将稀疏矩阵转化为稠密矩阵,onehot是独热编码后的结果。 另外一种方法是直接使用LabelBinarizer对文本进行独热编码。可以使用以下代码实现: from sklearn.preprocessing import LabelBinarizer encoder = LabelBinarizer() onehot = encoder.fit_transform(data['颜色']) print("独热编码结果:", onehot) 其中,data['颜色']表示待编码的分类特征列,onehot是独热编码后的结果。 最后一种方法是使用DictVectorizer。可以使用以下代码实现: from sklearn.feature_extraction import DictVectorizer encoder = DictVectorizer() onehot = encoder.fit_transform(data[['颜色']].to_dict('records')).toarray() print("独热编码结果:", onehot) 其中,data[['颜色']].to_dict('records')是将指定的特征列转化为字典形式,toarray()用于将稀疏矩阵转化为稠密矩阵,onehot是独热编码后的结果。 总结来说,在sklearn中进行独热编码,可以使用LabelEncoder和OneHotEncoder、LabelBinarizer或者DictVectorizer这几种方法。每一种方法都有其适用的情况,具体选择哪种方法取决于数据的特点和需求。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值