normalize hard sigmoid activation name with brulee

Author: EmilHvitfeldt

R/mlp.R

@@ -200,6 +200,7 @@ keras_mlp <-
         {.val {activation}}."
       )
     }
+    activation <- get_activation_fn(activation)
 
     if (penalty > 0 & dropout > 0) {
       cli::cli_abort("Please use either dropout or weight decay.", call = NULL)
@@ -351,7 +352,7 @@ mlp_num_weights <- function(p, hidden_units, classes) {
 }
 
 allowed_keras_activation <-
- c("elu", "exponential", "gelu", "hard_sigmoid", "linear", "relu", "selu", 
+ c("elu", "exponential", "gelu", "hardsigmoid", "linear", "relu", "selu", 
    "sigmoid", "softmax", "softplus", "softsign", "swish", "tanh")
 
 #' Activation functions for neural networks in keras
@@ -363,6 +364,13 @@ keras_activations <- function() {
   allowed_keras_activation
 }
 
+get_activation_fn <- function(arg, ...) {
+  if (arg == "hardsigmoid") {
+    arg <- "hard_sigmoid"
+  }
+  arg
+}
+
 ## -----------------------------------------------------------------------------
 
 #' @importFrom purrr map

tests/testthat/_snaps/mlp_keras.md

@@ -6,3 +6,13 @@
       Error:
       ! object 'novar' not found
 
+# all keras activation functions
+
+    Code
+      mlp(mode = "classification", hidden_units = 2, penalty = 0.01, epochs = 2,
+        activation = "invalid") %>% set_engine("keras", verbose = 0) %>% parsnip::fit(
+        Class ~ A + B, data = modeldata::two_class_dat)
+    Condition
+      Error in `parsnip::keras_mlp()`:
+      ! `activation` should be one of: elu, exponential, gelu, hardsigmoid, linear, relu, selu, sigmoid, softmax, softplus, softsign, swish, and tanh, not "invalid".
+