fit_fx_predictor {tidyhte} | R Documentation |
Fit a predictor for treatment effects
Description
This function predicts treatment effects in a second stage model.
Usage
fit_fx_predictor(full_data, weights, psi_col, ..., .pcate.cfg, .Model_cfg)
Arguments
full_data |
The full original data with all auxilliary columns. |
weights |
Weights to be used in the analysis. |
psi_col |
The unquoted column name of the calculated pseudo-outcome. |
... |
Covariate data, passed in as the unquoted names of columns in |
.pcate.cfg |
A |
.Model_cfg |
A |
Value
A list with two items:
-
model
- TheFX.Predictor
model object used internally for PCATE estimation. -
data
- The data augmented with column.pseudo_outcome_hat
for the cross-fit predictions of the HTE for each unit.
See Also
[Package tidyhte version 1.0.2 Index]