Merge branch 'main' into cox_boost_fs

mlr-org · Nov 29, 2023 · 681f2f7 · 681f2f7
2 parents 8c73521 + 4c891a0
commit 681f2f7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/R/learner_survivalmodels_surv_akritas.R b/R/learner_survivalmodels_surv_akritas.R
@@ -26,7 +26,9 @@ LearnerSurvAkritas = R6Class("LearnerSurvAkritas",
     initialize = function() {
       ps = ps(
         lambda = p_dbl(default = 0.5, lower = 0, upper = 1, tags = "predict"),
-        reverse = p_lgl(default = FALSE, tags = "train")
+        reverse = p_lgl(default = FALSE, tags = "train"),
+        ntime = p_dbl(default = 150, lower = 1, tags = "predict"),
+        round_time = p_int(default = 2, lower = 0, special_vals = list(FALSE), tags = "predict")
       )
 
       super$initialize(