predict.rkern: predict for rkern

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

Description

get estimated tau(x) using the trained rkern model

Usage

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

Arguments

object	an rkern 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)

rkern_fit = rkern(x, w, y)
rkern_est = predict(rkern_fit, x)

## End(Not run)

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