Return a DataFrame When Using a scikit-learn’s Transformer

Return a DataFrame When Using a scikit-learn’s Transformer

Applying a scikit-learn’s transformer on your DataFrame will return a NumPy array. If you want to return a pandas DataFrame instead, use SklearnTransformerWrapper along with your scikit-learn’s transformer.

This is a method of feature-engine.

My previous tips on tools for feature engineer.

Search

Scroll to Top

Work with Khuyen Tran

Work with Khuyen Tran