remove tuning parameter for rpart

topepo · topepo · commit 73adda690a73 · 2021-01-06T09:34:58.000-05:00
diff --git a/inst/add-in/parsnip_model_db.R b/inst/add-in/parsnip_model_db.R
@@ -52,6 +52,9 @@ get_tunable_param <- function(mode, package, model, engine) {
   if (model %in% c("bag_tree", "bag_mars")) {
     res <- res[0,]
   }
+  if (engine %in% c("rpart")) {
+    res <- res[res$parameter != "tree-depth",]
+  }
   res
 
 }

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,9 @@ get_tunable_param <- function(mode, package, model, engine) {`
`52`	`52`	`if (model %in% c("bag_tree", "bag_mars")) {`
`53`	`53`	`res <- res[0,]`
`54`	`54`	`}`
	`55`	`+ if (engine %in% c("rpart")) {`
	`56`	`+ res <- res[res$parameter != "tree-depth",]`
	`57`	`+ }`
`55`	`58`	`res`
`56`	`59`
`57`	`60`	`}`