1. เรียนรู้
    2. /
  2. Courses
    3. /
  3. MLflow 入門
Connected

Exercises

モデルのログと読み込み

Model API は、標準化された方法で MLflow Tracking から直接モデルをログ・読み込みできる仕組みを提供します。ML ライフサイクルにおいて、Model Engineering と Model Evaluation の段階でモデルとやり取りできることは重要です。

この演習では、Unicorn データセットを使って scikit-learn の Linear Regression モデルを作成します。モデルは MLflow Tracking にログし、そのアーティファクトをログした run_id を使って読み込みます。

はじめに、MLflow モジュールの scikit-learn ライブラリを使ってモデルをログします。続いて、run_id を用いて MLflow Tracking からモデルを読み込みます。

モデルは学習済みで、名前は lr_model です。

lr_model = LinearRegression()
lr_model.fit(X_train, y_train)

mlflow モジュールはインポート済みです。

คำแนะนำ

100 XP
  • アーティファクトパス "lr_tracking" の下にモデルを MLflow Tracking へログします。
  • 直近のランを表す変数 run を作成します。
  • 変数 run の run_id を取り出して、run_id という変数に設定します。
  • モデルをログしたアーティファクトパスと run_id を使ってモデルを読み込みます。