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.