Ordinal Encode Categorical Column

ordinal_encode_categorical_col.py

import pandas as pd
from sklearn.preprocessing import OrdinalEncoder

# Categorical columns
object_cols = [col for col in train_data.columns if train_data[col].dtype == "object"]

# valid_data에서의 value set이 train_data에서의 value set의 부분집합인 column들, train_data에 fit된 encoder를 사용하기위한 필요조건이다.
good_cols = [col for col in object_cols if set(valid_data[col]).issubset(set(train_data[col]))]

ordinal_encoder = OrdinalEncoder()
label_X_train[good_label_cols] = ordinal_encoder.fit_transform(train_data[good_label_cols])
label_X_valid[good_label_cols] = ordinal_encoder.transform(valid_data[good_label_cols])