mlr3proba (version 0.2.1)

mlr_graphs_survaverager: Survival Prediction Averaging Pipeline

Description

Wrapper around PipeOpSurvAvg to simplify Graph creation.

Usage

pipeline_survaverager(learners, param_vals = list(), graph_learner = FALSE)

Arguments

learners

(list()) List of LearnerSurvs to average.

param_vals

(list()) Parameters, including weights, to pass to PipeOpSurvAvg.

graph_learner

logical(1) If TRUE returns wraps the Graph as a GraphLearner otherwise (default) returns as a Graph.

Value

mlr3pipelines::Graph or mlr3pipelines::GraphLearner

See Also

Other pipelines: mlr_graphs_crankcompositor, mlr_graphs_distrcompositor, mlr_graphs_probregrcompositor, mlr_graphs_survbagging, mlr_graphs_survtoregr

Examples

# NOT RUN {
library("mlr3")
library("mlr3pipelines")

task = tgen("simsurv")$generate(10)
pipe = ppl(
  "survaverager",
  learners = lrns(c("surv.kaplan", "surv.coxph")),
  param_vals = list(weights = c(0.1, 0.9)),
  graph_learner = FALSE
 )
pipe$train(task)
pipe$predict(task)
# }