statsmodels.imputation.mice.MICEData.get_fitting_data

MICEData.get_fitting_data(vname)[source]

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 : str

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.