Compare model types with Hyperopt and MLflow

Note

The managed MLflow integration with Databricks on Google Cloud requires Databricks Runtime for Machine Learning 9.1 LTS or above.

This notebook demonstrates how to tune the hyperparameters for multiple models and arrive at a best model overall. It uses Hyperopt with SparkTrials to compare three model types, evaluating model performance with a different set of hyperparameters appropriate for each model type.

Compare models using scikit-learn, Hyperopt, and MLflow

Open notebook in new tab Copy link for import