statsmodels.treatment.treatment_effects.TreatmentEffect.aipw

TreatmentEffect.aipw(return_results=True, disp=False)[source]

ATE and POM from double robust augmented inverse probability weighting

Parameters:
return_resultsbool

If True, then a results instance is returned. If False, just ATE, POM0 and POM1 are returned.

dispbool

Indicates whether the scipy optimizer should display the optimization results

Returns:
TreatmentEffectsResults instance or tuple (ATE, POM0, POM1)

See also

TreatmentEffectsResults

Last update: Dec 16, 2024