LightGBMでどの特徴量がカテゴリ変数かを指定する

LightGBMにpd.DataFrameを渡す場合、SeriesのdtypeがcategoryだったらLightGBMもカテゴリ変数として扱ってくれる。 なので、前処理の時に以下のようにしてdtypeを変換するとよい。

train["Pclass"] = orig_train["Pclass"].astype("category")

参考：

blog.amedama.jp

www.tutorialspoint.com