Learn R Programming

SuperLearner (version 2.0-28.1)

predict.SuperLearner: Predict method for SuperLearner object

Description

Obtains predictions on a new data set from a SuperLearner fit. May require the original data if one of the library algorithms uses the original data in its predict method.

Usage

# S3 method for SuperLearner
predict(object, newdata, X = NULL, Y = NULL,
  onlySL = FALSE, ...)

Value

pred

Predicted values from Super Learner fit

library.predict

Predicted values for each algorithm in library

Arguments

object

Fitted object from SuperLearner

newdata

New X values for prediction

X

Original data set used to fit object, if needed by fit object.

Y

Original outcome used to fit object, if needed by fit object.

onlySL

Logical. If TRUE, only compute predictions for algorithms with non-zero coefficients in the super learner object. Default is FALSE (computes predictions for all algorithms in library).

...

Additional arguments passed to the predict.SL.* functions

Author

Eric C Polley epolley@uchicago.edu

Details

If newdata is omitted the predicted values from object are returned. Each algorithm in the Super Learner library needs to have a corresponding prediction function with the ``predict.'' prefixed onto the algorithm name (e.g. predict.SL.glm for SL.glm).

See Also

SuperLearner