日々精進

新しく学んだことを書き留めていきます

2016-12-29から1日間の記事一覧

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

AI

LightGBMにpd.DataFrameを渡す場合、SeriesのdtypeがcategoryだったらLightGBMもカテゴリ変数として扱ってくれる。 なので、前処理の時に以下のようにしてdtypeを変換するとよい。 train["Pclass"] = orig_train["Pclass"].astype("category") 参考: blog.a…