Machine learning workflows and Unity Catalog

Databricks Machine Learning features like Databricks Feature Store and Databricks AutoML can access data from Unity Catalog.

For an example notebook that shows how to train a machine-learning model that uses data in Unity Catalog and write predictions back to Unity Catalog, see Train a machine-learning model with Python from data in Unity Catalog.

To access data in Unity Catalog for machine learning workflows, you must use a Single User cluster. User Isolation clusters are not compatible with Databricks Runtime for Machine Learning.