使用 MLflow 實驗組織訓練回合

實驗是模型定型回合的組織單位。 實驗有兩種類型:工作區和筆記本。

  • 您可以從 Databricks 機器學習 UI 或 MLflow API 建立工作區實驗。 工作區實驗不會與任何筆記本相關聯,而且任何筆記本都可以使用實驗標識碼或實驗名稱來記錄這些實驗的執行。
  • 筆記本實驗與特定筆記本相關聯。 當您使用 mlflow.start_run() 啟動執行時,Azure Databricks 會自動建立筆記本實驗。

若要查看您有權存取的工作區中的所有實驗,請在提要字段中選取 [機器學習 > 實驗]。

實驗頁面

建立工作區實驗

本節說明如何使用 Azure Databricks UI 建立工作區實驗。 您可以直接從工作區或 [實驗] 頁面建立工作區實驗。

您也可以使用 MLflow APIDatabricks Terraform 提供者 搭配 databricks_mlflow_experiment

如需記錄執行至工作區實驗的指示,請參閱 記錄範例筆記本

  1. 按兩下 工作區圖示提要欄位中的 [工作區 ]。

  2. 流覽至您要在其中建立實驗的資料夾。

  3. 以滑鼠右鍵按兩下資料夾,然後選取 [ 建立 > MLflow 實驗]。

  4. 在 [建立 MLflow 實驗] 對話框中,輸入實驗的名稱和選擇性成品位置。 如果您未指定成品位置,則會將成品儲存在 中 dbfs:/databricks/mlflow-tracking/<experiment-id>

    Azure Databricks 支援 DBFS 和 Azure Blob 記憶體成品位置。

    若要將成品儲存在 Azure Blob 記憶體中,請指定格式 wasbs://<container>@<storage-account>.blob.core.windows.net/<path>的 URI。 儲存在 Azure Blob 記憶體中的成品不會出現在 MLflow UI 中;您必須使用 Blob 記憶體用戶端下載它們。

    注意

    當您將成品儲存在 DBFS 以外的位置時,成品不會出現在 MLflow UI 中。 儲存在 DBFS 以外的位置的模型無法在模型登錄中註冊。

  5. 按一下 [建立]。 空白實驗隨即出現。

您也可以從 [實驗] 頁面建立新的工作區實驗。 若要建立新的實驗,請使用 建立實驗下拉式清單 下拉功能表。 從下拉功能表中,您可以選取 AutoML 實驗 或空白(空白)實驗。

  • AutoML 實驗。 [設定 AutoML 實驗 ] 頁面隨即出現。 如需使用 AutoML 的詳細資訊,請參閱 使用 Azure Databricks AutoML UI 將 ML 模型定型。

  • 空白實驗。 [ 建立 MLflow 實驗 ] 對話框隨即出現。 在對話框中輸入名稱和選擇性成品位置,以建立新的工作區實驗。 預設成品位置為 dbfs:/databricks/mlflow-tracking/<experiment-id>

    若要記錄此實驗的執行,請使用實驗路徑呼叫 mlflow.set_experiment() 。 實驗路徑會出現在實驗頁面頂端。 如需詳細數據和範例筆記本,請參閱 記錄範例筆記本

建立筆記本實驗

當您在筆記本中使用 mlflow.start_run() 命令 時,執行會將計量和參數記錄至使用中實驗。 如果沒有實驗作用中,Azure Databricks 會建立筆記本實驗。 筆記本實驗會與其對應的筆記本共用相同的名稱和標識符。 筆記本標識碼是筆記本 URL 和標識碼結尾的數值標識碼。

或者,您可以將 Azure Databricks 工作區路徑傳遞至 mlflow.set_experiment() 中的現有筆記本,為其建立筆記本實驗。

如需記錄筆記本實驗的記錄執行指示,請參閱 記錄範例筆記本

注意

如果您使用 API 刪除筆記本實驗(例如, MlflowClient.tracking.delete_experiment() 在 Python 中),筆記本本身會移至 [垃圾桶] 資料夾。

檢視實驗

您有權存取的每個實驗都會出現在實驗頁面上。 您可以從此頁面檢視任何實驗。 按兩下實驗名稱以顯示實驗頁面。

存取實驗頁面的其他方式:

  • 您可以從工作區功能表存取工作區實驗的實驗頁面。
  • 您可以從筆記本存取筆記本實驗的實驗頁面。

若要搜尋實驗,請在 [篩選實驗 ] 欄位中輸入文字,然後按 Enter 或按兩下放大鏡圖示。 實驗清單會變更,只顯示那些在 [名稱]、[建立者]、[位置] 或 [描述] 數據行中包含搜尋文字的實驗。

點選數據表中任何實驗的名稱,以顯示其實驗頁面:

檢視實驗

實驗頁面會列出與實驗相關聯的所有執行。 從數據表中,您可以按兩下 實驗的 [執行名稱] 來開啟與實驗相關聯之任何回合的執行頁面。 [來源] 資料行可讓您存取建立執行的筆記本版本。 您也可以依計量或參數設定來搜尋和 篩選 執行。

檢視工作區實驗

  1. 按兩下 工作區圖示提要欄位中的 [工作區 ]。
  2. 移至包含實驗的資料夾。
  3. 按兩下實驗名稱。

檢視筆記本實驗

在筆記本的右側提要字段中,按兩下 [實驗] 圖示 實驗圖示

[實驗執行] 提要字段隨即出現,並顯示與筆記本實驗相關聯的每個回合摘要,包括執行參數和計量。 提要字段的頂端是筆記本最近記錄的實驗名稱(筆記本實驗或工作區實驗)。

檢視執行參數和計量

您可以從提要欄位瀏覽至實驗頁面,或直接流覽至執行。

  • 若要檢視實驗,請按兩下 外部連結 [實驗執行] 旁邊的最右邊。
  • 若要顯示 執行,請按兩下執行的名稱。

管理實驗

您可以從實驗頁面、 實驗頁面或工作區功能表,重新命名、刪除或管理您擁有之實驗的許可權。

注意

您無法直接重新命名、刪除或管理 Databricks Git 資料夾中筆記本所建立之 MLflow 實驗的許可權。 您必須在 Git 資料夾層級執行這些動作。

從實驗頁面或實驗頁面重新命名實驗

重要

這項功能處於公開預覽狀態

若要從實驗頁面或實驗頁面重新命名實驗,請按下 三個按鈕圖示 並選取 [ 重新命名]。

從工作區功能表重新命名實驗

  1. 按兩下 工作區圖示提要欄位中的 [工作區 ]。
  2. 移至包含實驗的資料夾。
  3. 以滑鼠右鍵按兩下實驗名稱,然後選取 [ 重新命名]。

複製實驗名稱

若要複製實驗名稱,請按兩下 複製圖示 實驗頁面頂端。 您可以在 MLflow 命令 set_experiment 中使用這個名稱來設定作用中的 MLflow 實驗。

實驗名稱圖示

您也可以從 筆記本中的實驗提要欄中複製實驗名稱。

刪除筆記本實驗

筆記本實驗是筆記本的一部分,無法個別刪除。 當您 刪除筆記本時,會刪除相關聯的筆記本實驗。 當您使用 UI 刪除筆記本實驗時,也會刪除筆記本。

若要使用 API 刪除筆記本實驗,請使用 工作區 API 來確保筆記本和實驗都會從工作區中刪除。

從工作區功能表刪除工作區實驗

  1. 按兩下 工作區圖示提要欄位中的 [工作區 ]。
  2. 移至包含實驗的資料夾。
  3. 以滑鼠右鍵按兩下實驗名稱,然後選取 [ 移至垃圾桶]。

從實驗頁面或實驗頁面刪除工作區或筆記本實驗

重要

這項功能處於公開預覽狀態

若要從實驗頁面或實驗頁面刪除實驗,請按下三個按鈕圖示並選取 [刪除]。

當您刪除筆記本實驗時,也會刪除筆記本。

變更實驗的許可權

若要從 實驗頁面變更實驗的許可權,請按兩下 [ 共用]。

實驗頁面許可權按鈕

您可以從實驗頁面變更您擁有之實驗的許可權。 按兩下 三個按鈕圖示 [ 動作] 資料行,然後選取 [ 許可權]。

如需實驗許可權等級的資訊,請參閱 MLFlow 實驗 ACL

在工作區之間複製實驗

若要在工作區之間移轉 MLflow 實驗,您可以使用社群驅動 開放原始碼 專案 MLflow Export-Import

使用這些工具,您可以:

  • 與相同或另一個追蹤伺服器中的其他數據科學家共用和共同作業。 例如,您可以將實驗從其他用戶複製到您的工作區。
  • 將 MLflow 實驗從本機追蹤伺服器複製到 Databricks 工作區中執行。
  • 將任務關鍵性實驗和模型備份到另一個 Databricks 工作區。