Ошибка значения: y содержит ранее невидимые метки: [8 9] ⇐ Python

[Anonymous]

У меня есть набор данных, который я использую label_encoding для предварительной обработки данных перед их передачей в модель.
Сначала я label_encoding все свои данные, а затем разделяю данные на обучение и тестирование.И я хочу использовать «label_encoder.inverse_transform», чтобы перевернуть код, но появляется следующее сообщение об ошибке:

Код: Выделить всё

y contains previously unseen labels: [8 9]

Сообщение об ошибке было ясным, но это меня смутило, поскольку перед разделением я закодировал все данные с помощью меток.

Код: Выделить всё

from sklearn.preprocessing import LabelEncoder
label_encoder = LabelEncoder()
category =[]
data_encoded = data_cleaned
for i in data_cleaned.columns:
if data_cleaned[i].dtype == 'category':
category.append(i)
data_encoded[i] = label_encoder.fit_transform(data_cleaned[i])

Разделение данных обучения и тестирования

Код: Выделить всё

target_variable = ['target']
data_encoded = data_encoded.dropna(subset=target_variable)
x = data_encoded.drop(target_variable, axis=1)
y = data_encoded[target_variable]

И попробуйте перевернуть закодированную метку

Код: Выделить всё

//only want to reverse the category column
for i in category:
X_test[i] = label_encoder.inverse_transform(X_test[i])
print(X_test[i])

Если бы кто-нибудь мог указать какое-то направление, это было бы здорово!

Подробнее здесь: https://stackoverflow.com/questions/788 ... labels-8-9