2019年5月6日月曜日

LightGBM

今日は LightGBM 。

Ubuntu では XGBoost を利用しているのですが、Win 環境では使えません。Win でも利用可能で、同等の精度、しかも速くなりそうな LightGBM に変更したいなあと考えていました。

で、実装。昨日の拡張です。
まずはLightGBMの組み込み。
次いでグリッドサーチ。
最後に ベイズ最適化を利用したアンサンブル。

ネットに情報があふれていますし、ライブラリが計算部分を担当してくれます。ブロックを組み立てる感覚なので、素人でもそれなりのツールを作ることができますね。

動かしてみたところ、LightGBM の計算は確かに速いと感じます。結果も上々(ROC はイマイチでしたが)。

最も良い結果は、感覚でハイパーパラメーターを調整した RF。
うーん。

0 件のコメント:

コメントを投稿