Get started with scikit-learn in Databricks

This 10-minute tutorial introduces you to machine learning in Databricks. It uses algorithms from the popular machine learning package scikit-learn along with MLflow for tracking the model development process and Hyperopt to automate hyperparameter tuning.

Requirements

Databricks Runtime ML

Example notebooks

Databricks recommends using MLflow autologging, illustrated in this notebook.

Get started with scikit-learn and MLflow autologging notebook

Open notebook in new tab

This notebook uses manual MLflow logging to track model development.

Get started with scikit-learn notebook

Open notebook in new tab