diff --git a/R/mlp.R b/R/mlp.R index a6616308f..1aa4d3c32 100644 --- a/R/mlp.R +++ b/R/mlp.R @@ -154,6 +154,8 @@ check_args.mlp <- function(object) { act_funs <- c("linear", "relu", "elu", "tanh") } else if (object$engine == "keras") { act_funs <- c("linear", "softmax", "relu", "elu") + } else if (object$engine == "h2o") { + act_funs <- c("relu", "tanh") } if (is.character(args$activation)) {