LightGBMでワインの品質データセットを識別してみる

wine

Kaggleで流行している、勾配ブースティング (Gradient Boosting) アルゴリズムを扱うためのフレームワークLightGBMの使い方を調べようと思い、scikit-learnに入っているワインの品質データセットの識別をやってみました。

このデータセットは３クラスのワインのデータが入っています。

環境構築

Mac OS Xで実行しました。

OpenMP のライブラリをbrewでインストール

brew install libomp

pipでパッケージをインストール

pip install lightgbm

ソースコード

LightGBMでパラメータから３種類のワインに分類する諸々参考にソースコードを作って見ました。

参考

blog.amedama.jp

yolo.love

参考になりました！ありがとうございます！

追伸

久々に更新したなー。