日々精進

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

LightGBMにはnative apiとscikit-learn apiがある

scikit-learn apiはnative apiのWrapper。メリデメは以下。

  • scikit-learn apiのメリット
    • scikit-learnのツールが使える。(Pipeline、cross_val_scoreなど)
    • scikit-learnの各アルゴリズムとIFが揃うので、アルゴリズムの取り替えが容易。これによってアンサンブルするときに前処理などのコードを再利用しやすくなる。

デメリットは今のところ分からないが、scikit-learn apiではできないことがあったりしそう。今のところ困ってないけど。。

参考:

stackoverflow.com