Example notebooks for Feature Engineering in Unity Catalog and Workspace Feature Store

This page includes example notebooks that illustrate the feature engineering workflow in 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

Open notebook in new tab

Use the following notebook if your workspace is not enabled for Unity Catalog.

Basic Workspace Feature Store example notebook

Open notebook in new tab

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

Open notebook in new tab

Use the following notebook if your workspace is not enabled for Unity Catalog.

Workspace Feature Store taxi example notebook

Open notebook in new tab

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

Open notebook in new tab

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 DynamoDB

Open notebook in new tab