Skip to content

Commit 6e0983f

Browse files
add tanh to allowed act_funs (#1240)
1 parent 1d052f2 commit 6e0983f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

R/mlp.R

+2-2
Original file line numberDiff line numberDiff line change
@@ -192,8 +192,8 @@ keras_mlp <-
192192
seeds = sample.int(10^5, size = 3),
193193
...) {
194194

195-
act_funs <- c("linear", "softmax", "relu", "elu")
196-
rlang::arg_match(activation, act_funs,)
195+
act_funs <- c("linear", "softmax", "relu", "elu", "tanh")
196+
rlang::arg_match(activation, act_funs)
197197

198198
if (penalty > 0 & dropout > 0) {
199199
cli::cli_abort("Please use either dropout or weight decay.", call = NULL)

0 commit comments

Comments
 (0)