From cb258e0702f9b5afdecf51f6282f5301b654e00a Mon Sep 17 00:00:00 2001 From: Emil Hvitfeldt Date: Wed, 29 Jan 2025 14:36:41 -0800 Subject: [PATCH] expand values_activation with more valid values --- R/param_activation.R | 7 ++++++- man/activation.Rd | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/R/param_activation.R b/R/param_activation.R index dc68f618..144c20d5 100644 --- a/R/param_activation.R +++ b/R/param_activation.R @@ -31,4 +31,9 @@ activation_2 <- function(values = values_activation) { #' @rdname activation #' @export -values_activation <- c("linear", "softmax", "relu", "elu", "tanh") +values_activation <- c("celu", "elu", "exponential", "gelu", "hardshrink", + "hardsigmoid", "hardtanh", "leaky_relu", "linear", "log_sigmoid", "relu", + "relu6", "rrelu", "selu", "sigmoid", "silu", "softmax", "softplus", + "softshrink", "softsign", "swish", "tanh", "tanhshrink" +) +# sort(unique(c(brulee::brulee_activations(), parsnip::keras_activations()))) \ No newline at end of file diff --git a/man/activation.Rd b/man/activation.Rd index 5912adff..60bf9865 100644 --- a/man/activation.Rd +++ b/man/activation.Rd @@ -7,7 +7,7 @@ \alias{values_activation} \title{Activation functions between network layers} \format{ -An object of class \code{character} of length 5. +An object of class \code{character} of length 23. } \usage{ activation(values = values_activation)