test: improve runtime

mlr-org · Sep 14, 2024 · 1d66f57 · 1d66f57
1 parent 6afb23d
commit 1d66f57
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 10 deletions.
diff --git a/tests/testthat/test_LearnerClassifAuto.R b/tests/testthat/test_LearnerClassifAuto.R
@@ -134,12 +134,10 @@ test_that("nnet works", {
 test_that("ranger works", {
   skip_on_cran()
   skip_if_not_installed("rush")
+  skip_if_not_installed("ranger")
   flush_redis()
 
   rush_plan(n_workers = 2)
-  skip_if_not_installed("ranger")
-
-
   task = tsk("penguins")
   learner = lrn("classif.auto",
     learner_ids = "ranger",
@@ -247,6 +245,7 @@ test_that("extra_trees and glmnet works", {
   task = tsk("penguins")
   learner = lrn("classif.auto",
     learner_ids = c("extra_trees", "glmnet"),
+    small_data_size = 1,
     resampling = rsmp("holdout"),
     measure = msr("classif.ce"),
     terminator = trm("evals", n_evals = 6)
@@ -268,6 +267,7 @@ test_that("lightgbm works", {
   learner = lrn("classif.auto",
     learner_ids = "lightgbm",
     lightgbm_eval_metric = "multi_logloss",
+    small_data_size = 1,
     resampling = rsmp("holdout"),
     measure = msr("classif.ce"),
     terminator = trm("evals", n_evals = 6)
@@ -291,10 +291,11 @@ test_that("xgboost, catboost and lightgbm work", {
     catboost_eval_metric = "MultiClass",
     lightgbm_eval_metric = "multi_logloss",
     xgboost_eval_metric = "mlogloss",
+    small_data_size = 1,
     resampling = rsmp("holdout"),
     lhs_size = 1,
     measure = msr("classif.ce"),
-    terminator = trm("evals", n_evals = 20),
+    terminator = trm("evals", n_evals = 7),
     callbacks = clbk("mlr3tuning.async_save_logs")
   )
 
@@ -311,10 +312,11 @@ test_that("all learner work", {
 
   task = tsk("penguins")
   learner = lrn("classif.auto",
-    small_data_size = 100,
+    small_data_size = 1,
+    lhs_size = 1,
+    resampling = rsmp("holdout"),
     measure = msr("classif.ce"),
-    terminator = trm("evals", n_evals = 20),
-    lhs_size = 1
+    terminator = trm("evals", n_evals = 20)
   )
 
   expect_class(learner$train(task), "LearnerClassifAuto")
@@ -366,7 +368,7 @@ test_that("large data set switch works", {
     large_data_size = 100,
     large_data_nthread = 4,
     large_data_learner_ids = "ranger",
-    small_data_size = 100,
+    small_data_size = 1,
     measure = msr("classif.ce"),
     terminator = trm("evals", n_evals = 1),
     lhs_size = 1,

diff --git a/tests/testthat/test_LearnerRegrAuto.R b/tests/testthat/test_LearnerRegrAuto.R
@@ -173,6 +173,7 @@ test_that("extra_trees and glmnet works", {
   task = tsk("mtcars")
   learner = lrn("regr.auto",
     learner_ids = c("extra_trees", "glmnet"),
+    small_data_size = 1,
     resampling = rsmp("holdout"),
     measure = msr("regr.rmse"),
     terminator = trm("evals", n_evals = 6)
@@ -189,6 +190,7 @@ test_that("lightgbm works", {
   task = tsk("mtcars")
   learner = lrn("regr.auto",
     learner_ids = "lightgbm",
+    small_data_size = 1,
     lightgbm_eval_metric = "multi_logloss",
     resampling = rsmp("holdout"),
     measure = msr("regr.rmse"),
@@ -206,6 +208,7 @@ test_that("xgboost, catboost and lightgbm work", {
   task = tsk("mtcars")
   learner = lrn("regr.auto",
     learner_ids = c("xgboost", "catboost", "lightgbm"),
+    small_data_size = 1,
     catboost_eval_metric = "MultiClass",
     lightgbm_eval_metric = "multi_logloss",
     xgboost_eval_metric = "mlogloss",
@@ -225,7 +228,8 @@ test_that("all learner work", {
 
   task = tsk("mtcars")
   learner = lrn("regr.auto",
-    small_data_size = 100,
+    small_data_size = 1,
+    resampling = rsmp("holdout"),
     measure = msr("regr.rmse"),
     terminator = trm("evals", n_evals = 20),
     lhs_size = 1
@@ -280,7 +284,8 @@ test_that("large data set switch works", {
     large_data_size = 100,
     large_data_nthread = 4,
     large_data_learner_ids = "ranger",
-    small_data_size = 100,
+    small_data_size = 1,
+    resampling = rsmp("holdout"),
     measure = msr("regr.rmse"),
     terminator = trm("evals", n_evals = 1),
     lhs_size = 1,