使用 Hyperopt 和 MLflow 比较模型类型

注意

不再维护 Hyperopt 的开源版本。

Hyperopt 将在下一个主要 DBR ML 版本中删除。 Azure Databricks 建议使用 Optuna 获得类似的体验,并访问更最新的超参数优化算法。

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

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

获取笔记本