register multinomial tunable arguments for brulee (#1055)

simonpcouch · web-flow · commit 5ba98ee826e8 · 2024-01-24T05:37:41.000-05:00
diff --git a/R/tunable.R b/R/tunable.R
@@ -270,7 +270,7 @@ tunable.multinomial_reg <- function(x, ...) {
     res$call_info[res$name == "mixture"] <-
       list(list(pkg = "dials", fun = "mixture", range = c(0.05, 1.00)))
   } else if (x$engine == "brulee") {
-    res <- add_engine_parameters(res, brulee_mlp_engine_args)
+    res <- add_engine_parameters(res, brulee_multinomial_engine_args)
   }
   res
 }

Original file line number	Diff line number	Diff line change
`@@ -270,7 +270,7 @@ tunable.multinomial_reg <- function(x, ...) {`
`270`	`270`	`res$call_info[res$name == "mixture"] <-`
`271`	`271`	`list(list(pkg = "dials", fun = "mixture", range = c(0.05, 1.00)))`
`272`	`272`	`} else if (x$engine == "brulee") {`
`273`		`- res <- add_engine_parameters(res, brulee_mlp_engine_args)`
	`273`	`+ res <- add_engine_parameters(res, brulee_multinomial_engine_args)`
`274`	`274`	`}`
`275`	`275`	`res`
`276`	`276`	`}`