databricks / mlops-stacks

This repo provides a customizable stack for starting new ML projects on Databricks that follow production best-practices out of the box.
https://docs.databricks.com/en/dev-tools/bundles/mlops-stacks.html
Apache License 2.0
460 stars 157 forks source link

Add Databricks Feature Store Component to MLOps Stacks #19

Closed vladimirk-db closed 1 year ago

vladimirk-db commented 2 years ago

This PR adds Databricks Feature Store support as a component in MLOps Stacks. It follows similar principles applied to training code support, including sample feature engineering code, automated unit and integration testing and infra-as-code setup for deployment.

It includes the following:

The sample code is based on the Feature Store Taxi Example Notebook.

This functionality is optional and guarded by a cookiecutter boolean flag (include_feature_store).

Tested:

TODOs (fixed):