当前位置:  开发笔记 > 编程语言 > 正文

LabelEncoder:如何保留显示原始变量和转换后变量的字典

如何解决《LabelEncoder:如何保留显示原始变量和转换后变量的字典》经验,为你挑选了1个好方法。



1> James King..:

我从创建了字典 classes_

le = preprocessing.LabelEncoder()
ids = le.fit_transform(labels)
mapping = dict(zip(le.classes_, range(len(le.classes_))))

去测试:

all([mapping[x] for x in le.inverse_transform(ids)] == ids)

应该回来True

之所以有效,是因为fit_transform用于numpy.unique同时计算标签编码和classes_属性:

def fit_transform(self, y):
    self.classes_, y = np.unique(y, return_inverse=True)
    return y

推荐阅读
围脖上的博博_771
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有