LabelEncoder

Кодирует категориальные метки целыми числами.

СигнатураLabelEncoder()

Превращает строковые или категориальные метки в целые числа от 0 до n-1. Предназначен в первую очередь для целевой переменной y, а не для признаков.

from sklearn.preprocessing import LabelEncoder

le = LabelEncoder()
y = le.fit_transform(['cat', 'dog', 'cat', 'bird'])
print(y)            # [1 2 1 0]
print(le.classes_)  # ['bird' 'cat' 'dog']
← Все записи: scikit-learn
Поддержать проект