- categorical_columns_small=['meal',
- 'distribution_channel',
- 'reserved_room_type',
- 'customer_type']
- categorical_columns_big=['arrival_date_month', 'country']
- encoder_s = OneHotEncoder(handle_unknown='ignore')
- encoder_b = MEstimateEncoder(handle_unknown='ignore')
- encoder_s.fit(data_train[categorical_columns_small])
- data_train_categorical_sm = pd.DataFrame(encoder.transform(data_train[categorical_columns_small]).toarray())
- data_test_categorical_sm = pd.DataFrame(encoder.transform(data_test[categorical_columns_small]).toarray())
- encoder_b.fit(data_train[categorical_columns_big])
- data_train_categorical_sm = pd.DataFrame(encoder.transform(data_train[categorical_columns_big]).toarray())
- data_test_categorical_sm = pd.DataFrame(encoder.transform(data_test[categorical_columns_big]).toarray())