機械学習モデルを保存した後のカテゴリ変数の扱い。

Question

機械学習モデルであるlightGBMを学習させ、pickleモジュールで保存・読み込みをしようとしています。
しかし、学習時に使用したデータにはカテゴリ変数が含まれており、LabelEncoderでint型にしました。

このモデルを保存して、未知のデータを予測するために読み込むときに、カテゴリ変数を変換させないといけません。

しかし、このカテゴリ変数の変換が学習時にした変換と同じになる確証はないと思うのですが、どうすればよろしいでしょうか？

Answer 1

モデル予測時に、質的データを適切に扱いたい

Comments

[Tommy_suzuki]

ご回答ありがとうございます。こちらのやり方で出来ました。

Answer 2

問題をちゃんと把握できているか自信ありませんが、学習時のエンコード結果を別に保存しておいて予測時にはそれを読み込んで利用するとよいのではないでしょうか。
参考：どんな場合でも同じラベルを付ける方法

Comments

[Tommy_suzuki]

ご回答ありがとうございます。こちらのやり方で出来ました。