Merge pull request #744 from tidymodels/check-args-h2o

qiushiyan · web-flow · commit a1791dd51b95 · 2022-06-02T17:34:27.000-05:00
Add available activation function in `mlp()` for h2o engine
diff --git 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)) {

Original file line number	Diff line number	Diff line change
`@@ -154,6 +154,8 @@ check_args.mlp <- function(object) {`
`154`	`154`	`act_funs <- c("linear", "relu", "elu", "tanh")`
`155`	`155`	`} else if (object$engine == "keras") {`
`156`	`156`	`act_funs <- c("linear", "softmax", "relu", "elu")`
	`157`	`+ } else if (object$engine == "h2o") {`
	`158`	`+ act_funs <- c("relu", "tanh")`
`157`	`159`	`}`
`158`	`160`
`159`	`161`	`if (is.character(args$activation)) {`