其他分享
首页 > 其他分享> > sklearn数据预处理函数LabelEncoder获取标签与编码映射

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