该函数可以把yes和no转化为0和1,或是把incident和normal转化为0和1。
标签都是非数字化的,所以我们需要对其进行转换。
from sklearn import preprocessing
labelList=['yes', 'no', 'no', 'yes']# 将标签矩阵二值化
lb = preprocessing.LabelBinarizer()#创建一个LabelBinarizer的实例lb
dummY=lb.fit_transform(labelList)#调用 lb 的fit_transform函数,将yes 和 no 转化成01
print(dummY)#打印结果
结果为:
[[1]
[0]
[0]
[1]]