predict.twostageSL: Predict method for twostageSL object

In wuziyueemory/twostageSL: Two-Stage Super Learner Prediction Function

Description

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

Usage

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

Arguments

object	Fitted object from twostageSL
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

Details

If newdata is omitted the predicted values from object are returned. Each algorithm in the library.2stage and library.1stage needs to have a corresponding prediction function with the “predict.” prefixed onto the algorithm name (e.g. predict.SL.glm for SL.glm).

Value

pred	Predicted values from twostageSL fit
library.predict	Predicted values for each algorithm in library

Author(s)

Ziyue Wu

See Also

twostageSL

wuziyueemory/twostageSL documentation built on Oct. 19, 2020, 3:45 p.m.