R/DefineLabelEncode.R

In houzeyu2683/PyKit:

##
##  Define label encode
DefineLabelEncode <- function(x){
  ##
  ##  Initial
  intNumber <- length(unique(x))
  strLabel  <- as.character(unique(x))
  intEncode <- seq(length(strLabel))
  listOutput <- list()
  listOutput[["Number"]] <- intNumber
  listOutput[["Label"]]  <- strLabel
  listOutput[["Encode"]] <- intEncode
  ##
  ##  Define encode function
  function(x){
    y = as.character(x)
    for(i in 1:listOutput[["Number"]]){
      y <- ifelse(y==listOutput[["Label"]][i], listOutput[["Encode"]][i], y) 
    }
    y[!(y %in% 1:listOutput[["Number"]])] <- listOutput[["Number"]] + 1
    y <- as.numeric(y)  
    return(y)
  }
}