64 votes

Pipeline Sklearn : Obtenez des noms de fonctionnalités après OneHotEncode dans ColumnTransformer

Je veux obtenir des noms de fonctionnalités après avoir adapté le pipeline.

alors

Après ajustement avec pandas dataframe, je peux obtenir des importances de fonctionnalité de

``

et j’ai `` essayé, mais j’ai eu une erreur

Comment puis-je obtenir des noms de fonctionnalités à partir de cela?

66voto

AILearning Points 535

Vous pouvez accéder à la feature_names en utilisant l’extrait suivant!

En utilisant sklearn >= version 0.21, nous pouvons le rendre plus simple:

Exemple reproductible :

Prograide.com

Prograide est une communauté de développeurs qui cherche à élargir la connaissance de la programmation au-delà de l'anglais.
Pour cela nous avons les plus grands doutes résolus en français et vous pouvez aussi poser vos propres questions ou résoudre celles des autres.

Powered by:

X