From ad9b181291f5c063ad2c30c5ed9af9a2449fa539 Mon Sep 17 00:00:00 2001 From: qiushi Date: Thu, 2 Jun 2022 17:17:14 -0500 Subject: [PATCH 1/2] add available act funs for h2o --- R/mlp.R | 2 ++ parsnip.Rproj | 1 - 2 files changed, 2 insertions(+), 1 deletion(-) 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)) { diff --git a/parsnip.Rproj b/parsnip.Rproj index a23ce7189..30e02be1a 100644 --- 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 From 93e3ef4e6da29d4c732426dd5ff6fe560e92fb15 Mon Sep 17 00:00:00 2001 From: qiushi Date: Thu, 2 Jun 2022 17:20:01 -0500 Subject: [PATCH 2/2] fix rproj --- parsnip.Rproj | 1 + 1 file changed, 1 insertion(+) diff --git a/parsnip.Rproj b/parsnip.Rproj index 30e02be1a..a23ce7189 100644 --- a/parsnip.Rproj +++ b/parsnip.Rproj @@ -17,5 +17,6 @@ StripTrailingWhitespace: Yes BuildType: Package PackageUseDevtools: Yes +PackageCleanBeforeInstall: Yes PackageInstallArgs: --no-multiarch --with-keep.source PackageRoxygenize: rd,collate,namespace