text_html_output_1.png
from sklearn.preprocessing import LabelEncoder

le = LabelEncoder()
y_train_prepared = le.fit_transform(y_train)

y_test_prepared = le.transform(y_test)
print(y_test_prepared.shape)