OptunaとMLflow
使用している機械学習ツール
Authored by tsugumi7878
June 29th 2021
こんにちは、北大機械学習サークルのtsugumi7878です。
ある程度機械学習モデルの開発が進んでくると、モデルのハイパーパラメーターチューニングや評価結果のログ取りが面倒になってきます。今回は最近使用し始めたツールについて簡単に書いておきます。
Optuna
ハイパーパラメータの推定に使用しています。今まで自分でパラメータを少しずつ変えて学習させてを手作業で繰り返し行っていました。それを自動でできるツールです。かなり楽になりました。
MLflow
評価結果のログに使用しています。Optunaの試行結果を保存しています。最終的にどのようなハイパーパラメータが良いのか、モデルごとの学習結果の比較も簡単に行えます。