Discover features and track feature lineage

With Databricks Feature Store, you can:

  • Search for feature tables by feature table name, feature, data source, or tag.

  • Control access to feature tables.

  • Identify the data sources used to create a feature table.

  • Identify models that use a particular feature.

  • Add a tag to a feature table.

  • Check feature freshness.

To access the Feature Store UI, in the sidebar, select Machine Learning > Feature Store. The Feature Store UI lists all of the available feature tables, along with the features in the table and the following metadata:

  • Who created the feature table.

  • Data sources used to compute the feature table.

  • Online stores where the feature table has been published.

  • Scheduled jobs that compute the features in the feature table.

  • The last time a notebook or job wrote to the feature table.

Feature store page

Search and browse for feature tables

Use the search box to search for feature tables. You can enter all or part of the name of a feature table, a feature, or a data source used for feature computation. You can also enter all or part of the key or value of a tag. Search text is case-insensitive.

Feature search example

Control access to feature tables

See Control access to feature tables.

Track feature lineage and freshness

In the UI you can track both how a feature was created and where it is used. For example, you can track the raw data sources, notebooks, and jobs that were used to compute the features. You can also track the online stores where the feature is published, the models trained with it, the serving endpoints that access it, and the notebooks and jobs that read it.

In the Feature Store UI, click the name of any feature table to display the feature table page.

On the feature table page, the Producers table provides information about all of the notebooks and jobs that write to this feature table so you can easily confirm the status of scheduled jobs and the freshness of the feature table.

producers table

The Features table lists all of the features in the table and provides links to the models, endpoints, jobs, and notebooks that use the feature.

features table

To return to the main Feature Store UI page, click Feature Store near the top of the page.

Add a tag to a feature table

Tags are key-value pairs that you can create and use to search for feature tables.

  1. On the feature table page, click Tag icon if it is not already open. The tags table appears.

    tag table
  2. Click in the Name and Value fields and enter the key and value for your tag.

  3. Click Add.

    add tag

Edit or delete a tag

To edit or delete an existing tag, use the icons in the Actions column.

tag actions