Added integer_encoder() function

Author: aldomann

NAMESPACE

@@ -1,6 +1,7 @@
 # Generated by roxygen2: do not edit by hand
 
 export("%>%")
+export(integer_encoder)
 export(multi_label_binariser)
 export(multi_label_binarizer)
 export(one_hot_encoder)

R/integer_encoder.R

@@ -0,0 +1,37 @@
+#' Integer Encoder
+#'
+#' @param data Input data frame
+#' @param feature Unquoted form of the feature/column to encode
+#'
+#' @return
+#' @export
+#'
+#' @examples
+#' set.seed(11)
+#' integer_encoder(iris[sample(1:150, 10),], Species)
+integer_encoder <- function(data, feature) {
+
+  # Process levels and labels
+  levels_raw <- data %>%
+    dplyr::pull({{ feature }}) %>%
+    base::as.character() %>%
+    base::unique() %>%
+    base::sort()
+
+  labels_raw <- base::seq(1, base::length(levels_raw), 1)
+
+  # Encoding
+  data <- data %>%
+    dplyr::mutate_at(
+      .vars = dplyr::vars({{ feature }}),
+      .funs = function(x) {
+        base::factor(
+          x,
+          levels = levels_raw,
+          labels = labels_raw
+        )
+      }
+    )
+
+  return(data)
+}