tabensemb.data.dataimputer.MiceLightgbmImputer#

class tabensemb.data.dataimputer.MiceLightgbmImputer(**kwargs)[source]#

Bases: AbstractImputer

An implementation of MICE with lightgbm.

Parameters:
**kwargs

Arguments for miceforest.ImputationKernel.mice

Methods

__init__(**kwargs)#

_defaults()

Defaults values for arguments defined in _cls_required_kwargs() and _required_kwargs()

_fit_transform(input_data, datamodule, **kwargs)

Fit the imputer and transform the input dataframe.

_transform(input_data, datamodule, **kwargs)

Transform the input tabular data using the fitted imputer.