fit_fx_predictor: Fit a predictor for treatment effects

In tidyhte: Tidy Estimation of Heterogeneous Treatment Effects

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 full_data
.pcate.cfg	A PCATE_cfg object describing what PCATEs to calculate (and how)
.Model_cfg	A Model_cfg object describing how the effect model should be estimated.

Value

A list with two items:

• model - The FX.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

Model_cfg, PCATE_cfg

tidyhte documentation built on Aug. 14, 2023, 5:08 p.m.