predict.SL.wang: Prediction method for 'SL.wang'

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

Description

Prediction method for SL.wang

Usage

## S3 method for class 'SL.wang'
predict(object, newdata, ...)

Arguments

object	An object of class SL.wang
newdata	A data.frame to generate predictions for
...	Other arguments (unused)

Value

A numeric vector of predictions

Examples

# load cost data
data(cost_data)
# fit wang model
fit_wang <- SL.wang(Y = cost_data$totalcost, X = cost_data[, c("female", "race")],
                              newX = cost_data[, c("female", "race")])
# get back predictions
pred_wang <- predict(fit_wang$fit, newdata = cost_data[,c("female", "race")])

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