Example notebooks for Feature Engineering in Unity Catalog and Workspace Feature Store
This page includes example notebooks that illustrate the feature engineering workflow in Azure Databricks for both Feature Engineering in Unity Catalog and Workspace Feature Store scenarios.
Note
With Databricks Runtime 13.3 LTS and above, any Delta table in Unity Catalog that has a primary key is automatically a feature table that you can use for model training and inference. When you use a table registered in Unity Catalog as a feature table, all Unity Catalog capabilities are automatically available to the feature table.
Basic feature engineering example
The basic feature engineering example notebook steps you through how to create a feature table, use it to train a model, and then perform batch scoring using automatic feature lookup. It also introduces you to the Feature Engineering UI and shows how you can use it to search for features and understand how features are created and used.
Basic Feature Engineering in Unity Catalog example notebook
Use the following notebook if your workspace is not enabled for Unity Catalog.
Basic Workspace Feature Store example notebook
Taxi example with point-in-time lookup
The taxi example notebook illustrates the process of creating features, updating them, and using them for model training and batch inference.
Feature Engineering in Unity Catalog taxi example notebook
Use the following notebook if your workspace is not enabled for Unity Catalog.
Workspace Feature Store taxi example notebook
Create Feature Serving endpoints
The following notebook illustrates how to use the Databricks SDK to create a Feature Serving endpoint using Databricks Online Tables.
Feature Serving example notebook with online tables
The following notebook illustrates how to use the Databricks SDK to create a Feature Serving endpoint using a third-party online store.
Feature Serving example notebook with CosmosDB
Feedback
https://aka.ms/ContentUserFeedback.
Coming soon: Throughout 2024 we will be phasing out GitHub Issues as the feedback mechanism for content and replacing it with a new feedback system. For more information see:Submit and view feedback for