開始使用免費開始

在專案執行中加入參數

在訓練時將參數作為變數傳入模型,可以用來設定模型的行為。這能讓你在不修改訓練程式碼本身的情況下,使用不同參數重複訓練模型數次。

在這個練習中,你會使用 mlflow projects 模組來執行一個用於訓練邏輯斯迴歸(Logistic Regression)模型的 Project,這是你保險實驗的一部分。你將撰寫使用 mlflow projects 模組的程式碼來執行你的專案,接著加入在訓練期間會作為超參數傳入模型的參數。

本練習屬於課程

MLflow 入門

檢視課程

練習說明

  • mlflow projects 模組呼叫 mlflow.projects.run() 函式。
  • 建立參數字典,並將 n_jobs_param 設為 2、fit_intercept_param 設為 False

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

import mlflow

# Set the run function from the MLflow Projects module
____.____.____(
    uri='./',
    entry_point='main',
    experiment_name='Insurance',
  	env_manager='local',
  	# Set parameters for n_jobs and fit_intercept
  	____={
        '____': ____, 
        '____': ____
    }
)
編輯並執行程式碼