始める無料で始める

モデルの保存と読み込み

Model API を使うと、同じ MLflow Tracking サーバーにアクセスできない開発者同士でも、ローカルファイルシステムを使ってモデルを共有できます。

この演習では、Unicorn データセットを使い、既存モデルから新しい LinearRegression モデルを学習します。最初に、ローカルファイルシステムから既存モデルを読み込みます。次に、その既存モデルを基に新しいモデルを学習し、ローカルファイルシステムに保存します。

既存モデルは "lr_local_v1" というディレクトリにローカル保存されています。mlflow モジュールはインポート済みです。

この演習はコースの一部です

MLflow 入門

コースを見る

演習の手順

  • MLflow モジュールの scikit-learn ライブラリを使って、ローカルファイルシステムのディレクトリ "lr_local_v1" からモデルを読み込みます。
  • mlflow モジュールの scikit-learn ライブラリを使って、モデルを "lr_local_v2" というディレクトリにローカル保存します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Load model from local filesystem
model = ____.____.____("____")

# Training Data
X = df[["R&D Spend", "Administration", "Marketing Spend", "State"]]
y = df[["Profit"]]
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size=0.7,random_state=0)
# Train Model
model.fit(X_train, y_train)

# Save model to local filesystem
____.____.____(____, "____")
コードを編集して実行