mlflow-quick-sample

概要

MLflow(Tracking)とHydraを動作させるサンプル。MLflowは別サーバ等は立てずにローカルで動作させることを想定。

パラメータやメトリクスをロギングしてMLflowのUIで確認するまでを実行する。

Install

python -m pip install -r requirements.txt

MLflow UIの起動

1. cd sample
2. mlflow ui --host 0.0.0.0 --port 5000 (UIの起動)
3. http://localhost:5000にアクセス

サンプルスクリプトの実行

1. cd sample
2. python mlflow_sample.py (基本的なAPIを利用したロギング)
3. python mlflow_lightgbm_sample.py (autologを利用したロギング)
4. python mlflow_lightgbm_hydra_sample.py (hydraを利用したロギング)
5. python mlflow_lightgbm_hydra_sample.py --multirun lightgbm.num_leaves=20,30,40 lightgbm.max_depth=5,7,9 (hydraのconfigパラメータを上書きして実行する場合)

Note

• lightgbmのデータにcategoryの列が含まれているとWARNING mlflow.lightgbm: Failed to infer model signatureが出力されるようである。

Reference

• MLflow Tracking
• MLflow Quickstart
• MLflowで実験管理入門
• Getting started | Hydra