statsmodels.treatment.treatment_effects.TreatmentEffect.from_data

classmethod TreatmentEffect.from_data(endog, exog, treatment, model='ols', **kwds)

Create models from data

Not yet implemented.

Parameters:

endogndarray

Outcome variable for the outcome model.

exogndarray

Explanatory variables for the outcome model.

treatmentndarray

Indicator array for observations with treatment (1) or without (0).

modelstr

Name of the model class to use for the outcome model.

**kwds

Additional keyword arguments passed to model classes.

Returns:

TreatmentEffect

Instance of the treatment effect class created from the data.