Специальное дерево решений

Question

У меня есть набор данных ниже, но его значения строковые.Поэтому, когда я хочу обучить дерево решений, эти функции должны быть преобразованы в num путем кодирования меток. Как я могу напечатать эти строки в дереве решений на этапе визуализации и при экспорте правил?

feature1  feature2  feature3  feature4  target
low       low       medium    high      sick
low       high      medium    medium    sick
high      medium    low       high      health
...

и это мой код для создания дерева решений:

le = preprocessing.LabelEncoder()
le.fit(data)
encoded_data = le.transform(data) 
tree.fit(encoded_data, target)

Answer 1

Распечатайте отображение классов:

list(le.classes_)

Выполните обратное преобразование меток в ваши категории:

list(le.inverse_transform(list_of_labels))

Документы здесь .

Вы также можете использовать панд .factorize () :

cat_list = ['a', 'b', 'c', 'd', 'd']
encoded_data, mapping_index = pd.Series(cat_list).factorize()