sklearn数据预处理函数LabelEncoder获取标签与编码映射
作者:互联网
文章目录
一、问题描述
利用sklearn中的LabelEncoder函数进行编码后,想要得到原来的标签和现在编码之间的映射关系。
le = LabelEncoder()
labels_en = le.fit_transform(labels).astype(np.int64)
二、问题解决
可以新建一个字典,用循环更新:
res = {}
for cl in le.classes_:
res.update({cl:le.transform([cl])[0]})
效果:
标签:编码,LabelEncoder,le,cl,labels,transform,预处理,sklearn 来源: https://blog.csdn.net/qq_21643227/article/details/116331024