R/label_count_encoding.R

In germayneng/DogEncode: Easy categorical encoding for machine learning feature engineering

#' @title label count encoding
#'
#' @examples
#'
#' \dontrun{
#'
#'
#' }
#' @import dplyr
#' @importFrom  data.table data.table
#' @export



label_count_encoding <- function(id){
  
  working_df <- data.table::data.table(name = id)
  working_df$name <- as.factor(working_df$name)
  summary_count <- table(id) %>% data.frame()
  colnames(summary_count) <- c("name","count")
  summary_count$name <- as.factor(summary_count$name)
  
  
  summary_count <- summary_count %>% arrange(count) %>%  mutate(encoded = row_number()) 
  working_df <- working_df %>% left_join(summary_count, by = "name")
  

  return(working_df$encoded)
  
}