scikit-learn と MLflow を使用したハイパーパラメーター チューニングの並列化

このノートブックでは、Hyperopt を使用してハイパーパラメーター チューニングの計算を並列化する方法を示します。 ここでは、SparkTrials クラスを使用して、クラスター ワーカー全体に計算を自動的に分散します。 また、後で結果を保存できるよう、Hyperopt 実行の自動 MLflow 追跡についても示します。

自動 MLflow 追跡を使用したハイパーパラメーター チューニングの並列化に関するノートブック

ノートブックを入手

ノートブックの最後のセルでアクションを実行すると、MLflow UI が表示されます。

Hyperopt MLflow のデモ