開始使用免費開始

儲存與載入模型

透過 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
____.____.____(____, "____")
編輯並執行程式碼