当前位置：网站首页>LabelEncoder和LabelBinarizer的区别

LabelEncoder和LabelBinarizer的区别

2022-08-11 05:35:00 【Pr4da】

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)