diff --git a/R/learner_aorsf_surv_aorsf.R b/R/learner_aorsf_surv_aorsf.R index 2bdee937c..b35da758b 100644 --- a/R/learner_aorsf_surv_aorsf.R +++ b/R/learner_aorsf_surv_aorsf.R @@ -37,11 +37,18 @@ delayedAssign( n_tree = p_int(default = 500L, lower = 1L, tags = "train"), n_split = p_int(default = 5L, lower = 1L, tags = "train"), n_retry = p_int(default = 3L, lower = 0L, tags = "train"), + n_thread = p_int(default = 1, lower = 0, tags = c("train", "predict")), + pred_aggregate = p_lgl(default = TRUE, tags = "predict"), mtry = p_int(default = NULL, lower = 1L, special_vals = list(NULL), tags = "train"), mtry_ratio = p_dbl(lower = 0, upper = 1, tags = "train"), + sample_with_replacement = p_lgl(default = TRUE, tags = "train"), + sample_fraction = p_dbl(lower = 0, upper = 1, default = .632, + tags = "train"), control_type = p_fct(levels = c("fast", "cph", "net"), default = "fast", tags = "train"), + split_rule = p_fct(levels = c("logrank", "cstat"), + default = "logrank", tags = "train"), control_fast_do_scale = p_lgl(default = TRUE, tags = "train"), control_fast_method = p_fct(levels = c("efron", "breslow"), default = "efron", tags = "train"), diff --git a/tests/testthat/test_paramtest_aorsf_surv_aorsf.R b/tests/testthat/test_paramtest_aorsf_surv_aorsf.R index 669e085b5..8765b02c1 100644 --- a/tests/testthat/test_paramtest_aorsf_surv_aorsf.R +++ b/tests/testthat/test_paramtest_aorsf_surv_aorsf.R @@ -38,6 +38,7 @@ test_that("paramtest surv.aorsf predict", { "pred_horizon", # all times required for distr "pred_type", # handled internally "na_action", # not implemented + "verbose_progress", # handled internally "boundary_checks" # not implemented ) paramtest = run_paramtest(learner, fun_list, exclude, tag = "predict")