Google
Cloud Platform
Amazon Web Services
Microsoft AzureWorkspace Feature Store (legacy)
Note
This documentation covers the Workspace Feature Store. Workspace Feature Store is available only for workspaces created before August 19, 2024, 4:00:00 PM (UTC).
Databricks recommends using Feature Engineering in Unity Catalog. Workspace Feature Store will be deprecated in the future.
Why use Workspace Feature Store?
Workspace Feature Store is fully integrated with other components of Databricks.
Discoverability. The Feature Store UI, accessible from the Databricks workspace, lets you browse and search for existing features.
Lineage. When you create a feature table in Databricks, the data sources used to create the feature table are saved and accessible. For each feature in a feature table, you can also access the models, notebooks, jobs, and endpoints that use the feature.
Integration with model scoring and serving. When you use features from Feature Store to train a model, the model is packaged with feature metadata. When you use the model for batch scoring or online inference, it automatically retrieves features from Feature Store. The caller does not need to know about them or include logic to look up or join features to score new data. This makes model deployment and updates much easier.
Point-in-time lookups. Feature Store supports time series and event-based use cases that require point-in-time correctness.
How does Workspace Feature Store work?
The typical machine learning workflow using Feature Store follows this path:
Write code to convert raw data into features and create a Spark DataFrame containing the desired features.
Write the DataFrame as a feature table in the Workspace Feature Store.
Train a model using features from the feature store. When you do this, the model stores the specifications of features used for training. When the model is used for inference, it automatically joins features from the appropriate feature tables.
Register model in Model Registry.
You can now use the model to make predictions on new data. The model automatically retrieves the features it needs from Feature Store.
Start using Workspace Feature Store
To get started, try these example notebooks. The basic 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.
The taxi example notebook illustrates the process of creating features, updating them, and using them for model training and batch inference.