From 7fe1385a0e43b1518d98ad5bfa90d254e5dc7e81 Mon Sep 17 00:00:00 2001 From: Raphael Sonabend Date: Tue, 28 Nov 2023 16:29:21 +0000 Subject: [PATCH] Update learner_survivalmodels_surv_akritas.R --- R/learner_survivalmodels_surv_akritas.R | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/R/learner_survivalmodels_surv_akritas.R b/R/learner_survivalmodels_surv_akritas.R index 461fc46dc..20b391ed8 100644 --- 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(