Automatisk funktionssökning med MLflow-modeller på Databricks

Modellservern kan automatiskt söka efter funktionsvärden från publicerade onlinebutiker eller från onlinetabeller. I den här artikeln beskrivs hur du arbetar med onlinebutiker. Information om hur du arbetar med onlinetabeller finns i Använda onlinetabeller för funktionsservering i realtid.

Krav

  • Modellen måste ha loggats med FeatureEngineeringClient.log_model (för funktionsutveckling i Unity Catalog) eller FeatureStoreClient.log_model (för Arbetsytans funktionslager krävs v0.3.5 och senare).
  • Onlinebutiken måste publiceras med skrivskyddade autentiseringsuppgifter.

Kommentar

Du kan publicera funktionstabellen när som helst före modelldistributionen, inklusive efter modellträningen.

Automatisk funktionsökning

Azure Databricks Model Serving har stöd för automatisk funktionssökning från dessa onlinebutiker:

  • Azure Cosmos DB (v0.5.0 och senare)

Automatisk funktionssökning stöds för följande datatyper:

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

Åsidosätta funktionsvärden i poängsättning av onlinemodeller

Alla funktioner som krävs av modellen (loggas med FeatureEngineeringClient.log_model eller FeatureStoreClient.log_model) söks automatiskt upp från onlinebutiker för modellbedömning. Om du vill åsidosätta funktionsvärden vid bedömning av en modell med hjälp av ett REST-API med modellservern inkluderar du funktionsvärdena som en del av API-nyttolasten.

Kommentar

De nya funktionsvärdena måste överensstämma med funktionens datatyp som förväntat av den underliggande modellen.

Notebook-exempel: Unity Catalog

Med Databricks Runtime 13.3 LTS och senare kan alla Delta-tabeller i Unity Catalog med en primärnyckel användas som en funktionstabell. När du använder en tabell som är registrerad i Unity Catalog som en funktionstabell är alla Unity Catalog-funktioner automatiskt tillgängliga för funktionstabellen.

Den här exempelanteckningsboken visar hur du publicerar funktioner i en onlinebutik och sedan servar en tränad modell som automatiskt söker efter funktioner från onlinebutiken.

OnlineButik med Unity Catalog-exempelanteckningsbok

Hämta notebook-fil

Notebook-exempel: Funktionsarkiv för arbetsyta

Den här exempelanteckningsboken visar hur du publicerar funktioner i en onlinebutik och sedan servar en tränad modell som automatiskt söker efter funktioner från onlinebutiken.

Exempelanteckningsbok för Online Store

Hämta notebook-fil