モデルの保存と読み込み
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
____.____.____(____, "____")