From 1ee485b5d1b128f1de10912f5eea96bf7a92bd11 Mon Sep 17 00:00:00 2001 From: Emil Hvitfeldt Date: Tue, 28 Jan 2025 14:23:23 -0800 Subject: [PATCH] add tanh to allowed act_funs --- R/mlp.R | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/R/mlp.R b/R/mlp.R index 5080c555e..4ea677e09 100644 --- a/R/mlp.R +++ b/R/mlp.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)