モデル推論の例
このノートブックでは、「 MLflow を使用した scikit-learn モデル トレーニングの追跡」で説明されている糖尿病データセットでトレーニングされた ElasticNet モデルを使用します。 このノートブックでは、次の方法を示します。
- MLflow エクスペリメント UI を使用してデプロイするモデルを選択する
- トレーニング済みモデルをscikit-learnモデルとして読み込みます
- モデルから PySpark UDF を作成する
- UDF を適用して、予測列を データフレーム に追加します