bnnSurvival (version 0.1.5)

predict,bnnSurvivalBaseLearner-method: Compute prediction for all samples.

Description

Compute prediction for all samples.

Usage

# S4 method for bnnSurvivalBaseLearner
predict(object, train_data, test_data,
  timepoints, metric, weighting_function, k)

Arguments

object

bnnSurvivalBaseLearner object

train_data

Training data (with response)

test_data

Test data (without response)

timepoints

Timepoint to predict at

metric

Metric used

weighting_function

Weighting function used

k

Number of nearest neighbors