使用 Hyperopt 和 MLflow 比较模型类型

注意

不再维护 Hyperopt 的开源版本。

Hyperopt 将不再预安装在 Databricks Runtime ML 17.0 及更高版本上。 Azure Databricks 建议改用 Optuna,以便获得类似的体验,并访问最新的超参数优化算法。

此笔记本演示如何优化多个模型的超参数并得到一个总体最佳的模型。 它结合使用 Hyperopt 与 SparkTrials 来比较三种模型类型,并为每种模型类型使用一组不同的超参数来评估模型性能。

使用 scikit-learn、Hyperopt 和 MLflow 来比较模型

获取笔记本