This page provides two example notebooks that illustrate using 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