在 Databricks 上使用 MLflow 模型的自動功能查閱

模型服務 可以自動從已發佈的在線商店或在線數據表查閱功能值。 本文說明如何使用在線商店。 如需使用在線數據表的相關信息,請參閱 使用在線數據表進行即時功能服務

需求

  • 模型必須已記錄 FeatureEngineeringClient.log_model 為 (針對 Unity 目錄中的功能工程)或 FeatureStoreClient.log_model (針對工作區功能存放區,需要 v0.3.5 和更新版本)。
  • 必須以唯讀認證發佈在線商店

注意

您可以在模型部署前隨時發佈功能數據表,包括模型定型之後。

自動功能查閱

Azure Databricks 模型服務 支援從這些在線商店進行自動功能查閱:

  • Azure Cosmos DB (v0.5.0 和更新版本)

下列資料類型支援自動功能查閱:

  • IntegerType
  • FloatType
  • BooleanType
  • StringType
  • DoubleType
  • LongType
  • TimestampType
  • DateType
  • ShortType
  • DecimalType
  • ArrayType
  • MapType

覆寫在線模型評分中的特徵值

模型所需的所有功能(使用 FeatureEngineeringClient.log_modelFeatureStoreClient.log_model記錄)都會從在線商店自動查閱以取得模型評分。 若要在使用 REST API 搭配 模型服務 評分模型時覆寫特徵值,請在 API 承載中包含特徵值。

注意

新的特徵值必須符合基礎模型預期的功能數據類型。

筆記本範例:Unity 目錄

使用 Databricks Runtime 13.3 LTS 和更新版本時,Unity 目錄中具有主鍵的任何 Delta 數據表都可以當做功能數據表使用。 當您使用在 Unity 目錄註冊為功能資料表的數據表時,功能數據表會自動提供所有 Unity 目錄功能。

此範例筆記本說明如何將功能發佈至在線商店,然後提供定型的模型,以自動從在線商店查閱功能。

使用 Unity 目錄範例筆記本的在線商店

取得筆記本

筆記本範例:工作區功能存放區

此範例筆記本說明如何將功能發佈至在線商店,然後提供定型的模型,以自動從在線商店查閱功能。

在線商店範例筆記本

取得筆記本