Estimate the Outcome Mechanism

Estimate the Outcome Mechanism

joint_stoch_shift_est_Q(
  exposures,
  deltas,
  mu_learner,
  covars,
  av,
  at,
  outcome_type
)

Arguments

exposures

A character vector of exposures to be shifted.

deltas

A numeric indicating the magnitude of the shift to be computed for the exposure A. This is passed to the internal shift_additive and is currently limited to additive shifts.

mu_learner

Object containing a set of instantiated learners from the sl3, to be used in fitting an ensemble model.

covars

A character vector covariates to adjust for.

av

A dataframe of validation data specific to the fold

at

A dataframe of training data specific to the fold

outcome_type

Variable type of the outcome

Value

A data.table with two columns, containing estimates of the outcome mechanism at the natural value of the exposure Q(A, W) and an upshift of the exposure Q(A + delta, W).

Details

Compute the outcome regression for the observed data, including with the shift imposed by the intervention. This returns the outcome regression for the observed data (at A) and under the counterfactual shift shift (at A + delta).