日々精進

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

LightGBMのtrainデータにカテゴリ変数を含めると「ValueError: could not convert string to float」エラー

原因はカテゴリ変数の場合は数値に変換してからLightGBMに渡さないといけないため。 trainメソッドにはcategorical_featureという引数があり、これでカテゴリ変数がどれか指定するがその値は以下の制約がある。

・int32の範囲の数値 ・0から連続した数値になっていること ・マイナスの値はすべて欠損値として扱われる

参考: lightgbm.readthedocs.io