estimate_ite_aipw: Estimate the Individual Treatment Effect (ITE) using...

In CRE: Interpretable Discovery and Inference of Heterogeneous Treatment Effects

Estimate the Individual Treatment Effect (ITE) using Augmented Inverse Probability Weighting (AIPW)

Description

Estimates the Individual Treatment Effect using Augmented Inverse Probability Weighting given a response vector, a treatment vector, a features matrix, an estimation model for the propensity score and estimation model for the outcome.

Usage

estimate_ite_aipw(y, z, X, learner_ps = "SL.xgboost", learner_y = "SL.xgboost")

Arguments

y	An observed response vector.
z	A treatment vector.
X	A features matrix.
learner_ps	A estimation model for the propensity score.
learner_y	A estimation model for the outcome.

Value

A list of ITE estimates.

CRE documentation built on Oct. 19, 2024, 5:07 p.m.