Skip to content

Commit 3d7a514

Browse files
committed
get back the old hyper parameters and set some non-zero trimming threshold
1 parent 9536dda commit 3d7a514

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

tests/testthat/helper-05-ml-learner.R

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -178,8 +178,8 @@ get_default_mlmethod_irm = function(learner) {
178178
mlmethod_g = paste0("regr.", learner))
179179

180180
params = list(
181-
params_g = list(cp = 0.02, minsplit = 20),
182-
params_m = list(cp = 0.02, minsplit = 20))
181+
params_g = list(cp = 0.01, minsplit = 20),
182+
params_m = list(cp = 0.01, minsplit = 20))
183183

184184
}
185185
ml_g = mlr3::lrn(mlmethod$mlmethod_g)
@@ -214,9 +214,9 @@ get_default_mlmethod_iivm = function(learner) {
214214
mlmethod_r = paste0("classif.", learner))
215215

216216
params = list(
217-
params_m = list(cp = 0.02, minsplit = 20),
218-
params_g = list(cp = 0.02, minsplit = 20),
219-
params_r = list(cp = 0.02, minsplit = 20))
217+
params_m = list(cp = 0.01, minsplit = 20),
218+
params_g = list(cp = 0.01, minsplit = 20),
219+
params_r = list(cp = 0.01, minsplit = 20))
220220

221221
}
222222
ml_g = mlr3::lrn(mlmethod$mlmethod_g)

tests/testthat/test-double_ml_iivm.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,14 +10,14 @@ if (on_cran) {
1010
learner = "rpart",
1111
dml_procedure = "dml2",
1212
score = "LATE",
13-
trimming_threshold = c(0),
13+
trimming_threshold = c(0.0001),
1414
stringsAsFactors = FALSE)
1515
} else {
1616
test_cases = expand.grid(
1717
learner = "cv_glmnet",
1818
dml_procedure = c("dml1", "dml2"),
1919
score = "LATE",
20-
trimming_threshold = c(0),
20+
trimming_threshold = c(0.0001),
2121
stringsAsFactors = FALSE)
2222
}
2323

tests/testthat/test-double_ml_irm_user_score.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,15 +25,15 @@ if (on_cran) {
2525
learner = "regr.rpart",
2626
learner_m = "classif.rpart",
2727
dml_procedure = "dml2",
28-
trimming_threshold = 0,
28+
trimming_threshold = 0.0001,
2929
stringsAsFactors = FALSE)
3030
test_cases[".test_name"] = apply(test_cases, 1, paste, collapse = "_")
3131
} else {
3232
test_cases = expand.grid(
3333
learner = "regr.glmnet",
3434
learner_m = "classif.glmnet",
3535
dml_procedure = c("dml1", "dml2"),
36-
trimming_threshold = c(0, 0.01),
36+
trimming_threshold = c(0.0001, 0.01),
3737
stringsAsFactors = FALSE)
3838
test_cases[".test_name"] = apply(test_cases, 1, paste, collapse = "_")
3939
}

0 commit comments

Comments
 (0)