LabelEncoder和LabelBinarizer的区别

LabelBinarizer

import numpy as np
from sklearn.preprocessing import LabelBinarizer

lb = LabelBinarizer()
a = np.array([1,2,3,2,5,6,6])
b = lb.fit_transform(a)
print(b)

输出为：

[[1 0 0 0 0]
 [0 1 0 0 0]
 [0 0 1 0 0]
 [0 1 0 0 0]
 [0 0 0 1 0]
 [0 0 0 0 1]
 [0 0 0 0 1]]

LabelBinarizer它将标签转换成了one-hot的形式。

LabelEncoder

import numpy as np
from sklearn.preprocessing import LabelBinarizer

le = LabelEncoder()
a = np.array([1,2,3,2,5,6,6])
e = le.fit_transform(a)
print(e)

输出为：

[0 1 2 1 3 4 4]

LabelEncoder()对标签进行了编码，转换成了对应数值。

本文原载于我的简书