一、问题描述

利用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]})

效果:
在这里插入图片描述

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐