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

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

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

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

コメントを投稿

0 コメント