LabelEncoder()不会存储参数?

时间:2017-08-31 14:29:20

标签: python scikit-learn

LabelEncoder不会“记住”参数。当我使用它来拟合和转换数据然后询问参数时,我得到的只是{}。这使得无法在新数据上重复使用编码器。

示例:

from sklearn.preprocessing import LabelEncoder

encode = LabelEncoder()
encode.fit_transform(['one', 'two', 'three'])
print(encode.get_params())

不确定预期的格式,但我希望有{['one', 0], ['two', 1], ['three', 2]}

之类的内容

实际结果:{}

我在:

Darwin-16.7.0-x86_64-i386-64bit
Python 3.6.1 |Anaconda 4.4.0 (x86_64)| (default, May 11 2017, 13:04:09) 
[GCC 4.2.1 Compatible Apple LLVM 6.0 (clang-600.0.57)]
NumPy 1.12.1
SciPy 0.19.0
Scikit-Learn 0.18.1

1 个答案:

答案 0 :(得分:3)

标签编码器将参数存储在 classes _ 属性中。您可以获取转换这些类并创建字典的编码值。此编码器将使用具有相同标签的新数据,否则将引发 ValueError 。在要编码的标签上使用转换方法。

from sklearn import preprocessing

encode = preprocessing.LabelEncoder()
encode.fit_transform(['one', 'two', 'three'])
keys = encode.classes_
values = encode.transform(encode.classes_)
dictionary = dict(zip(keys, values))
print(dictionary)

输出:{'three': 1, 'two': 2, 'one': 0}