Feature engineering with scikit-learn

The example notebook on this page illustrates how to use scikit-learn on Databricks for feature engineering.

Use scikit-learn with MLflow integration on Databricks

This notebook shows a complete end-to-end example of loading data, training a model, distributed hyperparameter tuning, and model inference. It also illustrates how to use MLflow and the model registry.

Note

The following notebook may include functionality that is not available in this release of Databricks on Google Cloud.

Use scikit-learn with MLflow integration on Databricks

Open notebook in new tab Copy link for import