predict.rlasso: predict for rlasso

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

Description

get estimated tau(x) using the trained rlasso model

Usage

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

Arguments

object	an rlasso object
newx	covariate matrix to make predictions on. If null, return the tau(x) 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)
w = rbinom(n, 1, 0.5)
y = pmax(x[,1], 0) * w + x[,2] + pmin(x[,3], 0) + rnorm(n)

rlasso_fit = rlasso(x, w, y)
rlasso_est = predict(rlasso_fit, x)

## End(Not run)

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