enable parsnip to work with functions wit parameterized labels

Author: ‘topepo’

R/extract.R

@@ -113,7 +113,7 @@ extract_parameter_set_dials.model_spec <- function(x, ...) {
 eval_call_info <-  function(x) {
   if (!is.null(x)) {
     # Look for other options
-    allowed_opts <- c("range", "trans", "values")
+    allowed_opts <- c("range", "trans", "values", "label")
     if (any(names(x) %in% allowed_opts)) {
       opts <- x[names(x) %in% allowed_opts]
     } else {

R/tunable.R

@@ -199,8 +199,8 @@ brulee_mlp_engine_args <-
     ~name,                                                  ~call_info,
     "momentum",      list(pkg = "dials", fun = "momentum", range = c(0.5, 0.95)),
     "batch_size",      list(pkg = "dials", fun = "batch_size", range = c(3, 10)),
-    "hidden_units_2",                  list(pkg = "dials", fun = "hidden_units"),
-    "activation_2",                      list(pkg = "dials", fun = "activation"),
+    "hidden_units_2",                  list(pkg = "dials", fun = "hidden_units", label = "# Hidden Units (layer 2)"),
+    "activation_2",                      list(pkg = "dials", fun = "activation", label = "Activation Function (layer 2)"),
     "stop_iter",                          list(pkg = "dials", fun = "stop_iter"),
     "class_weights",                  list(pkg = "dials", fun = "class_weights"),
     "decay",                             list(pkg = "dials", fun = "rate_decay"),
@@ -364,6 +364,7 @@ tunable.mlp <- function(x, ...) {
     res$call_info[res$name == "epochs"] <-
       list(list(pkg = "dials", fun = "epochs", range = c(5L, 500L)))
   } else if (x$engine == "brulee_two_layer") {
+    rlang::check_installed("brulee", version = "0.3.0.9000")
     res <- add_engine_parameters(res, brulee_mlp_engine_args)
     res$call_info[res$name == "learn_rate"] <-
       list(list(pkg = "dials", fun = "learn_rate", range = c(-3, -1/2)))