Added option to choose factor or integer as final data type on integer_encoder() function

Author: aldomann

DESCRIPTION

@@ -1,6 +1,6 @@
 Package: encoder
 Title: Data Encoding Methods for Categorical Features
-Version: 2.0.0
+Version: 2.1.0
 Authors@R: c(
     person(given = "Alfredo",
            family = "Hernández",

R/integer_encoder.R

@@ -2,14 +2,15 @@
 #'
 #' @param data Input data frame
 #' @param feature Unquoted form of the feature/column to encode
+#' @param as_factor Whether the factors should be converted to factors or pure integer
 #'
 #' @return
 #' @export
 #'
 #' @examples
 #' set.seed(11)
 #' integer_encoder(iris[sample(1:150, 10),], Species)
-integer_encoder <- function(data, feature) {
+integer_encoder <- function(data, feature, as_factor = TRUE) {
 
   # Process levels and labels
   levels_raw <- data %>%
@@ -33,5 +34,13 @@ integer_encoder <- function(data, feature) {
       }
     )
 
+  if(!as_factor) {
+    data <- data %>%
+      dplyr::mutate_at(
+        .vars = dplyr::vars({{ feature }}),
+        .funs = as.integer
+      )
+  }
+
   return(data)
 }