add available act funs for h2o

qiushiyan · qiushiyan · commit ad9b181291f5 · 2022-06-02T17:17:14.000-05:00
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)) {
diff --git a/parsnip.Rproj b/parsnip.Rproj
@@ -17,6 +17,5 @@ StripTrailingWhitespace: Yes
 
 BuildType: Package
 PackageUseDevtools: Yes
-PackageCleanBeforeInstall: Yes
 PackageInstallArgs: --no-multiarch --with-keep.source
 PackageRoxygenize: rd,collate,namespace

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)) {`