skrub-data / skrub

Prepping tables for machine learning

Home Page:https://skrub-data.org/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

`get_feature_names_out` returns lists instead of numpy arrays

Vincent-Maladiere opened this issue · comments

Problem Description

This issue is very minor. In scikit-learn, calling get_feature_names_out from any estimator returns a numpy array, while in skrub it returns a list. It might be worth bringing consistency between the two libraries by outputting numpy arrays in skrub and to limit the potential confusion for our users.
What do @skrub-data/devs think?

Feature Description

This issue suggests returning a numpy array instead of a list in get_feature_names_out of all skrub estimators that implement it.

Alternative Solutions

No response

Additional Context

No response