search

MIDASverse / MIDASpy

Python package for missing-data imputation with deep learning
Apache License 2.0
137 stars 37 forks source link

Scikit-learn interface #12

Open kaiserdan opened 3 years ago

kaiserdan commented 3 years ago

It's usual for imputation or data pre-processing packages in python to support the scikit-learn interface. The interface allows the library to be used in data pipelines and existing scikit-learn infrastructures. Is there any plan on explicitly supporting it in the library?

References: https://scikit-learn.org/stable/modules/impute.html https://scikit-learn.org/stable/developers/develop.html https://scikit-learn.org/stable/data_transforms.html

tsrobinson commented 3 years ago

Thanks for bringing this to our attention. We agree that it would be great to have support for the scikit-learn interface, and so we do plan to support this in the near-term -- likely as a new sub module.