本文共 395 字，大约阅读时间需要 1 分钟。

自己动手的第一个机器学习算法

from sklearn import preprocessingfeature = [[0,1], [1,1], [0,0], [1,0]]label= ['yes', 'no', 'yes', 'no']lb = preprocessing.LabelBinarizer() #构建一个转换对象Y = lb.fit_transform(label)re_label = lb.inverse_transform(Y)print(Y)print(re_label)

输出

[[1] [0] [1] [0]]['yes' 'no' 'yes' 'no']

转换函数将字符串label进行数字化，数字范围从0开始，并且将label转换为了一个列向量。最后在ML任务完成之后，输出的时候需要还原之前的label，因此使用函数inverse_transform()。

转载地址：http://awyki.baihongyu.com/