predict.cvboost: predict for cvboost

In xnie/rlearner: R-learner for Heterogeneous Treatment Effect Estimation

Description

predict for cvboost

Usage

## S3 method for class 'cvboost'
predict(object, newx = NULL, ...)

Arguments

object	a cvboost object
newx	covariate matrix to make predictions on. If null, return the predictions on the training data
...	additional arguments (currently not used)

Value

vector of predictions

Examples

## Not run: 
n = 100; p = 10

x = matrix(rnorm(n*p), n, p)
y = pmax(x[,1], 0) + x[,2] + pmin(x[,3], 0) + rnorm(n)

fit = cvboost(x, y, objective="reg:squarederror")
est = predict(fit, x)

## End(Not run)

xnie/rlearner documentation built on April 11, 2021, 12:49 a.m.