ordinal encoder function

ordinal_encoder.py

def EncodeCategoricalData(train_df, 
                          test_df, 
                          cols=ordinal_encode_cols):
  # funtion to create ordinal encoder object and
  # assigns -1 to the unseen labels of the test set
  # returns ordinal encoded train and test datasets
  # respectively
  ordinal_encoder = OrdinalEncoder(cols=ordinal_encode_cols,
                                   return_df=True,
                                   handle_unknown="value",
                                   handle_missing="return_nan")
  ordinal_encoder.fit(train_df)
    
  ord_encoded_train = (ordinal_encoder
                       .transform(train_df))
    
  ord_encoded_test = (ordinal_encoder
                      .transform(test_df))

  return ord_encoded_train, ord_encoded_test