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.

Use scikit-learn with MLflow integration on Databricks

Open notebook in new tab