1
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 1 year has passed since last update.

機械学習で一度40個以上のモデルをトレーニングできるライブラリ:lazypredict

Posted at

先週github上で「lazypredict」という面白いライブラリを見つけました。
これを使うと、データがすべてのモデル上でのトレーニング結果を表示することができます。

Colab上で「カリフォルニア住宅価格データ」を使って価格予測の回帰トレーニング試してみました
10-26-1.png

モデルのトレーニング結果は以下となります。一部だけのスクリーンショットですが、全部では約40個のモデルの結果が表示されます。
10-26-2.png

回帰問題なのでデフォルトの評価指標はR-SquaredとRMSEですね。分類問題にすると混同行列になります。
また、"custom_metric"というパラメーター使えばカスタマイズ評価指標の結果も出せるようになります。
あとは処理時間(TimeTaken)も見れます。

複数モデルのために、ハイパーパラメーターチューニングはできないですが、データマイニングのケースではハイパーパラメーターチューニングもそれほど重要ではないため、確認用としては特に問題ないと思います。
もちろんモデルに関しては、比較せずLightGBMにしても良いのですが、資料作成の時に「こういう比較結果ありますよ」みたいな感じで活用できるではないかと思います。

興味ある方試してみてください。
github: https://github.com/shankarpandala/lazypredict

1
1
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
1
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?