sklearn.preprocessing.LabelBinarizer() 标签二值化

该函数可以把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]]