Skip to content

Commit 3f3910f

Browse files
expand values_activation with more valid values (#371)
1 parent b56e9bc commit 3f3910f

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

R/param_activation.R

+6-1
Original file line numberDiff line numberDiff line change
@@ -31,4 +31,9 @@ activation_2 <- function(values = values_activation) {
3131

3232
#' @rdname activation
3333
#' @export
34-
values_activation <- c("linear", "softmax", "relu", "elu", "tanh")
34+
values_activation <- c("celu", "elu", "exponential", "gelu", "hardshrink",
35+
"hardsigmoid", "hardtanh", "leaky_relu", "linear", "log_sigmoid", "relu",
36+
"relu6", "rrelu", "selu", "sigmoid", "silu", "softmax", "softplus",
37+
"softshrink", "softsign", "swish", "tanh", "tanhshrink"
38+
)
39+
# sort(unique(c(brulee::brulee_activations(), parsnip::keras_activations())))

man/activation.Rd

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)