今日は LightGBM 。
Ubuntu では XGBoost を利用しているのですが、Win 環境では使えません。Win でも利用可能で、同等の精度、しかも速くなりそうな LightGBM に変更したいなあと考えていました。
で、実装。昨日の拡張です。
まずはLightGBMの組み込み。
次いでグリッドサーチ。
最後に ベイズ最適化を利用したアンサンブル。
ネットに情報があふれていますし、ライブラリが計算部分を担当してくれます。ブロックを組み立てる感覚なので、素人でもそれなりのツールを作ることができますね。
動かしてみたところ、LightGBM の計算は確かに速いと感じます。結果も上々(ROC はイマイチでしたが)。
最も良い結果は、感覚でハイパーパラメーターを調整した RF。
うーん。
0 件のコメント:
コメントを投稿