Получение списка кодированных меток ярлыков в sklearn?

Question

У меня есть ярлык, закодированный моим набором данных, который содержит около 100 различных строк.

Итак, есть ли способ получить список меток вдоль исходных строк.

Как пример:

apple --> 3 
orange --> 4 
pear --> 1 

Answer 1

Из документации sklearn.preprocessing.LabelEncoder:

classes_: array of shape (n_class,) Holds the label for each class.

Построение dict от этого должно быть тривиальным.

Как правило, этап sklearn обычно сохраняет этот тип информации в некотором члене, заканчивающемся знаком подчеркивания (например, classes_ в этом случае).

Пример

from sklearn import preprocessing 
>>> preprocessing.LabelEncoder().fit(['apply', 'banana']).classes_ 
array(['apply', 'banana'], 
     dtype='|S6')