Train and register machine learning models with Unity Catalog

To get started, import the following notebook.

Open notebook in new tab

To import the notebook:

Next to the notebook, click Copy link for import.

In your workspace, click Workspace.

Next to a folder, click , then click Import

Click URL, then paste in the link you copied.

The imported notebook appears in the folder you selected. Double-click the notebook name to open it.

At the top of the notebook, select your Databricks Machine Learning cluster to attach the notebook to it.

The notebook is divided into several high-level sections:

Setup.

Read data from CSV files and writing it to Unity Catalog.

Load the data into Pandas dataframes and clean it up.

Train a basic classification model.

Tune hyperparameters and optimize the model.

Register the model in Unity Catalog.

Write the results to a new table and share it with other users.

To run a cell, click Run. To run the entire notebook, click Run All.