python - scikit-learnのワンホットエンコーディングからカテゴリ機能をバックトラックしますか?

Question

LASSO 回帰予測モデルを構築しようとしています。scikit-learn で OneHotEncoder を使用して、one-hot aka one-of-K スキームを使用して、すべてのカテゴリ整数機能をエンコードしました。結果に基づくと、51 個のパラメーターのみが実際に予測モデルに影響を与えます。これらのパラメータを調査したいのですが、上記のようにエンコードされています。ホットエンコードされた配列に対応するカテゴリ整数機能を抽出する方法を知っていますか? ありがとう！

Answer 1

sklearn の Pipeline と FeatureUnion を強化するために ple を設計しました。これにより、ワンホット エンコーディングやその他の前処理ステップの後にカテゴリ機能をバックトラックすることもできます。さらに、GraphX による変換を「描画」できます。たとえば、

ple は、私の Github ページで見つけることができます。