statsmodels.imputation.mice.MICEData.get_fitting_data

MICEData.get_fitting_data(vname)

Return the data needed to fit a model for imputation.

The data is used to impute variable vname, and therefore only includes cases for which vname is observed.

Values of type PatsyFormula in init_kwds or fit_kwds are processed through Patsy and subset to align with the model’s endog and exog.

Parameters:

vname : string

The variable for which the fitting data is returned.

Returns:

endog : DataFrame

Observed values of vname.

exog : DataFrame

Regression design matrix for imputing vname.

init_kwds : dict-like

The init keyword arguments for vname, processed through Patsy as required.

fit_kwds : dict-like

The fit keyword arguments for vname, processed through Patsy as required.