predict_causal_forest: Estimates conditional average treatment effects based on a...
In till-tietz/rcf: Heterogenous Treatment Effect Estimation Using Random Forests

predict_causal_forest

R Documentation

Estimates conditional average treatment effects based on a trained causal forest and test data

Estimates conditional average treatment effects based on a trained causal forest and test data

predict_causal_forest(data, cf, predict_oob = TRUE)

`data`	a data frame with observations to predict and no NA
`cf`	output of the rcf causal_forest function
`predict_oob`	set to TRUE if you want to predict cates on the observations you used to fit the causal forest (uses out of bag observations); set to FALSE if you wish to use a test data set. Default TRUE.