toggle mlp activation by engine

[EmilHvitfeldt]

library(tidymodels)

model <- 
  mlp(activation = tune()) %>%
  set_engine("keras") %>%
  set_mode("classification")

set.seed(1)
res <- tune_grid(
  workflow(class ~ ., model),
  vfold_cv(sim_classification(1000), v = 2),
  grid = 10
)

res |>
  collect_metrics() |>
  count(activation)
#> # A tibble: 10 × 2
#>    activation       n
#>    <chr>        <int>
#>  1 elu              3
#>  2 exponential      3
#>  3 hard_sigmoid     3
#>  4 linear           3
#>  5 selu             3
#>  6 sigmoid          3
#>  7 softmax          3
#>  8 softplus         3
#>  9 swish            3
#> 10 tanh             3

model <- 
  mlp(activation = tune()) %>%
  set_engine("brulee") %>%
  set_mode("classification")

set.seed(1)
res <- tune_grid(
  workflow(class ~ ., model),
  vfold_cv(sim_classification(1000), v = 2),
  grid = 10
)
#> → A | warning: Current loss in NaN. Training wil be stopped.
#> There were issues with some computations   A: x1
#> There were issues with some computations   A: x2
#> There were issues with some computations   A: x2
#> 

res |>
  collect_metrics() |>
  count(activation)
#> # A tibble: 10 × 2
#>    activation      n
#>    <chr>       <int>
#>  1 elu             3
#>  2 hardshrink      3
#>  3 hardtanh        3
#>  4 leaky_relu      3
#>  5 log_sigmoid     3
#>  6 relu6           3
#>  7 selu            3
#>  8 softplus        3
#>  9 softsign        3
#> 10 tanhshrink      3

[github-actions]

This pull request has been automatically locked. If you believe you have found a related problem, please file a new issue (with a reprex: https://reprex.tidyverse.org) and link to this issue.