predict.skern: predict for skern

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

Description

get estimated tau(x) using the trained skern model

Usage

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

Arguments

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

skern_fit = skern(x, w, y)
skern_est = predict(skern_fit, x)

## End(Not run)

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