turn default engine to "torch"

mlverse · Feb 9, 2023 · a5a8764 · a5a8764
1 parent 44c35eb
commit a5a8764
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/NEWS.md b/NEWS.md
@@ -23,6 +23,9 @@
 * `tabnet_explain()` is now correctly handling missing values in predictors. (#77)
 * `dataloader` can now use `num_workers>0` (#83)
 * new default values for `batch_size` and `virtual_batch_size` do not limit performance on mid-range devices.
+* add default `engine="torch"` to tabnet parsnip model (#114)
+* fix `autoplot()` warnings turned into errors with {ggplot2} v3.4 (#113)
+
 
 # tabnet 0.3.0
 

diff --git a/R/parsnip.R b/R/parsnip.R
@@ -269,7 +269,7 @@ tabnet <- function(mode = "unknown", epochs = NULL, penalty = NULL, batch_size =
 
   # Save some empty slots for future parts of the specification
   out <- list(args = args, eng_args = NULL,
-              mode = mode, method = NULL, engine = NULL)
+              mode = mode, method = NULL, engine = "torch")
 
   # set classes in the correct order
   class(out) <- parsnip::make_classes("tabnet")