Merge branch 'main' into poisson-linear-reg

Author: topepo

DESCRIPTION

@@ -25,7 +25,7 @@ Imports:
     ggplot2,
     globals,
     glue,
-    hardhat (>= 1.4.0.9002),
+    hardhat (>= 1.4.0.9003),
     lifecycle,
     magrittr,
     pillar,

NAMESPACE

@@ -377,6 +377,7 @@ importFrom(generics,tune_args)
 importFrom(generics,varying_args)
 importFrom(ggplot2,autoplot)
 importFrom(glue,glue_collapse)
+importFrom(hardhat,contr_one_hot)
 importFrom(hardhat,extract_fit_engine)
 importFrom(hardhat,extract_fit_time)
 importFrom(hardhat,extract_parameter_dials)

R/contr_one_hot.R

@@ -192,8 +192,8 @@ keras_mlp <-
            seeds = sample.int(10^5, size = 3),
            ...) {
 
-    act_funs <- c("linear", "softmax", "relu", "elu")
-    rlang::arg_match(activation, act_funs,)
+    act_funs <- c("linear", "softmax", "relu", "elu", "tanh")
+    rlang::arg_match(activation, act_funs)
 
     if (penalty > 0 & dropout > 0) {
       cli::cli_abort("Please use either dropout or weight decay.", call = NULL)

R/mlp.R

@@ -30,6 +30,10 @@ generics::augment
 #' @export
 generics::required_pkgs
 
+#' @importFrom hardhat contr_one_hot
+#' @export
+hardhat::contr_one_hot
+
 #' @importFrom hardhat extract_spec_parsnip
 #' @export
 hardhat::extract_spec_parsnip