predict.cv.causalBoosting: Make predictions from a fitted cross-validated causal...

In saberpowers/causalLearning: Methods for heterogeneous treatment effect estimation

Description

Make predictions from a fitted cross-validated causal boosting model

Usage

## S3 method for class 'cv.causalBoosting'
predict(object, newx, newtx = NULL,
  type = c("treatment.effect", "conditional.mean", "response"),
  num.trees = object$num.trees.min.effect, naVal = 0, ...)

Arguments

object	a fitted cv.causalBoosting object
newx	matrix of new covariates for which estimated treatment effects are desired
newtx	option vector of new treatment assignments (only used if type = 'individual')
type	type of prediction required: 'treatment.effect' returns estimated treatment effect. 'conditional.mean' returns two predictions, one for each arm. 'response' returns prediction for arm corresponding to newtx.
num.trees	number of shallow causal trees to use for prediction
naVal	value with which to replace NA predictions
...	ignored

Value

a vector or matrix of predictions corresponding to the rows of newx

saberpowers/causalLearning documentation built on May 30, 2019, 8:26 a.m.