text_html_output_1.png
from sklearn.preprocessing import LabelEncoder

from sklearn.preprocessing import LabelEncoder
le = LabelEncoder()
le.fit(y_train)

y_train_processed = le.transform(y_train)
y_test_processed = le.transform(y_test)

le.inverse_transform([0, 1])