未知データに対するダミー変数化

Question

pythonで機械学習モデルを作成する際に、カテゴリ変数を

get_dummies

でダミー変数化しますが、モデル作成後、
推論したいデータのカテゴリ変数を同様のルールでダミー変数化する
よい方法はありますでしょうか。

よろしくお願いします。

Answer 1

get_dummiesは使わず、sklearnで以下のようなモデルを用いてダミー変数化すれば良いです。

sklearn.preprocessing.OrdinalEncoder — scikit-learn 0.20.0 documentation
sklearn.preprocessing.LabelEncoder — scikit-learn 0.20.0 documentation
sklearn.preprocessing.OneHotEncoder — scikit-learn 0.20.0 documentation

OrdinalEncoder or LabelEncoderとOneHotEncoderを組み合わせてPipelineとすることで、意図しているような機能が達成できるかと思います。

Comments

[yuyu127]

hayataka2049様
ありがとうございました。
いただいた情報で調べてみます。