Merge pull request #286 from tidymodels/check_sparse_arg

add and use check_sparse_arg()

Author: EmilHvitfeldt

R/aaa.R

@@ -30,6 +30,12 @@ table0 <- function(x) {
   purrr::set_names(res$n, res$tokens)
 }
 
+check_sparse_arg <- function(x) {
+  if (!is.null(x)) {
+    rlang::arg_match0(x, c("auto", "yes", "no"), arg_nm = "sparse")
+  }
+}
+
 sparse_is_yes <- function(x) {
   !is.null(x) && x == "yes"
 }

R/dummy_hash.R

@@ -180,6 +180,7 @@ prep.step_dummy_hash <- function(x, training, info = NULL, ...) {
   check_bool(x$signed, arg = "signed")
   check_number_whole(x$num_terms, min = 0, arg = "num_terms")
   check_bool(x$collapse, arg = "collapse")
+  check_sparse_arg(x$sparse)
 
   check_type(training[, col_names], types = c("string", "factor", "ordered"))
 

R/texthash.R

@@ -169,6 +169,7 @@ prep.step_texthash <- function(x, training, info = NULL, ...) {
   check_bool(x$signed, arg = "signed")
   check_number_whole(x$num_terms, min = 0, arg = "num_terms")
   check_string(x$prefix, arg = "prefix")
+  check_sparse_arg(x$sparse)
 
   check_type(training[, col_names], types = "tokenlist")
 

R/tf.R

@@ -188,6 +188,7 @@ prep.step_tf <- function(x, training, info = NULL, ...) {
   check_number_decimal(x$weight, arg = "weight")
   check_character(x$vocabulary, allow_null = TRUE, arg = "vocabulary")
   check_string(x$prefix, arg = "prefix")
+  check_sparse_arg(x$sparse)
 
   check_type(training[, col_names], types = "tokenlist")
 

R/tfidf.R

@@ -175,6 +175,7 @@ prep.step_tfidf <- function(x, training, info = NULL, ...) {
   rlang::arg_match0(x$norm, c("l1", "l2", "none"), arg_nm = "norm")
   check_bool(x$sublinear_tf, arg = "sublinear_tf")
   check_string(x$prefix, arg = "prefix")
+  check_sparse_arg(x$sparse)
 
   check_type(training[, col_names], types = "tokenlist")