fix: default prediction type should not be 'distr' but crank for pipe…

…lines
mlr-org · Feb 10, 2024 · 7f584e6 · 7f584e6
1 parent 12bbc61
commit 7f584e6
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/R/pipelines.R b/R/pipelines.R
@@ -31,7 +31,6 @@ pipeline_survaverager = function(learners, param_vals = list(), graph_learner =
 
   if (graph_learner) {
     gr = mlr3pipelines::GraphLearner$new(gr)
-    gr$predict_type = "distr"
   }
 
   gr
@@ -98,7 +97,6 @@ pipeline_survbagging = function(learner, iterations = 10, frac = 0.7, avg = TRUE
 
     if (graph_learner) {
       gr = mlr3pipelines::GraphLearner$new(gr)
-      gr$predict_type = "distr"
     }
 
     gr
@@ -159,7 +157,6 @@ pipeline_crankcompositor = function(learner,
 
   if (graph_learner) {
     gr = mlr3pipelines::GraphLearner$new(gr)
-    gr$predict_type = "distr"
   }
 
   gr