R/apply_factor_encoding.R

In agilebean/catcoding: Compare Categorical Encoding Algorithms

################################################################################
#
# Script:  R/apply_factor_encoding.R
# Output:  training.set, testing.set - no encoding
#
################################################################################
apply_factor_encoding <- function(
  encoding, training_original, testing_original, target_label) {
  
  # set feature labels
  features.original <- training_original %>% select(-target_label) %>% names
  
  return(
    list(
      features.labels = features.original,
      target.label = target_label,
      testing.set = testing_original,
      training.set = training_original
    )
  )
}